From aee3c2846f6aa1b7e99c7af16fd762d2fd13b87a Mon Sep 17 00:00:00 2001 From: selmanozleyen Date: Wed, 3 Jul 2024 16:58:03 +0200 Subject: [PATCH] fix --- tests/transforms/test_add_edge_index.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/transforms/test_add_edge_index.py b/tests/transforms/test_add_edge_index.py index 0993dc5..fdcda8b 100644 --- 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())