scipy

xorbitsai · Nov 5, 2024 · a6f5c54 · a6f5c54
1 parent 17d41ba
commit a6f5c54
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/xorbits/_mars/tensor/reduction/tests/test_reduction_execution.py b/python/xorbits/_mars/tensor/reduction/tests/test_reduction_execution.py
@@ -604,7 +604,8 @@ def test_count_nonzero_execution(setup):
     t = count_nonzero(arr, axis=0)
 
     res = t.execute().fetch()
-    expected = np.count_nonzero(raw.todense(), axis=0)
+    # TODO sparse matrix does not support .A in higher versions
+    expected = np.count_nonzero(raw.A, axis=0)
     np.testing.assert_equal(res, expected)
 
     t = count_nonzero(arr, axis=1)