Merge pull request #55 from theislab/fix/toarray

Fix Sparse ".A" Not Found
theislab · Jul 3, 2024 · 402a0a8 · 402a0a8
2 parents 8ea77ca + aee3c28
commit 402a0a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/transforms/test_add_edge_index.py b/tests/transforms/test_add_edge_index.py
@@ -78,4 +78,4 @@ def test_add_edge_index():
     adata = tf(adata)
     assert torch.equal(adata.uns["edge_index"], edge_index_gt)
     assert torch.allclose(adata.uns["edge_weight"], edge_weight_gt)
-    assert np.allclose(adata.obsp["pred_distances"].A, adata.obsp["gt_distances"].A)
+    assert np.allclose(adata.obsp["pred_distances"].toarray(), adata.obsp["gt_distances"].toarray())