Fix assertion in test_label_transfer

scverse · Sep 24, 2024 · 3316fca · 3316fca
1 parent 07f7488
commit 3316fca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/tools/_perturbation_space/test_simple_perturbation_space.py b/tests/tools/_perturbation_space/test_simple_perturbation_space.py
@@ -250,4 +250,4 @@ def test_label_transfer():
     assert all(adata.obs["perturbation_transfer_certainty"] <= 1)
     assert not all(adata.obs["perturbation_transfer_certainty"] == 1)
     is_known = adata.obs["perturbation"] != "unknown"
-    assert all(adata.loc[is_known, "perturbation_transfer_certainty"] == 1)
+    assert all(adata.obs.loc[is_known, "perturbation_transfer_certainty"] == 1)