Explicit cast from matrix to array

flatironinstitute · Jul 2, 2024 · 5d60893 · 5d60893
1 parent 2cb7753
commit 5d60893
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/sparse_dot_mkl/_mkl_interface/__init__.py b/sparse_dot_mkl/_mkl_interface/__init__.py
@@ -90,7 +90,15 @@ def _validate_dtype():
                 output_type='csr_matrix'
             )
 
-        if not _np.allclose(test_comparison, final_array.todense()):
+            final_array = final_array.todense()
+
+            # Make sure this is an array and not np.matrix
+            try:
+                final_array = final_array.A
+            except AttributeError:
+                pass
+
+        if not _np.allclose(test_comparison, final_array):
             raise ValueError("Match failed after matrix conversion")
         _destroy_mkl_handle(csr_ref)
     finally: