We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 8ea77ca + aee3c28 commit 402a0a8Copy full SHA for 402a0a8
1 file changed
tests/transforms/test_add_edge_index.py
@@ -78,4 +78,4 @@ def test_add_edge_index():
78
adata = tf(adata)
79
assert torch.equal(adata.uns["edge_index"], edge_index_gt)
80
assert torch.allclose(adata.uns["edge_weight"], edge_weight_gt)
81
- 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())
0 commit comments