last

Author: sayandip18

sparse/_common.py

@@ -149,9 +149,7 @@ def tensordot(a, b, axes=2, *, return_type=None):
 
 @njit
 def check_nan(a, b, fv1, fv2):
-    x = a.data
-    y = b.data
-    return np.isnan(fv1) or np.isnan(fv2) or np.isnan(np.min(x)) or np.isnan(np.min(y))
+    return np.isnan(fv1) or np.isnan(fv2) or np.isnan(np.min(a)) or np.isnan(np.min(b))
 
 
 def matmul(a, b):
@@ -177,7 +175,7 @@ def matmul(a, b):
     numpy.matmul : NumPy equivalent function.
     COO.__matmul__ : Equivalent function for COO objects.
     """
-    if check_nan(a, b, a.fill_value, b.fill_value):
+    if check_nan(a.data, b.data, a.fill_value, b.fill_value):
         warnings.warn("Warning: nan is not propagated in matrix multiplication")
     check_zero_fill_value(a, b)
     if not hasattr(a, "ndim") or not hasattr(b, "ndim"):