Merge pull request #49 from YosefLab/canergen-patch-2

Update test_models.py
YosefLab · Jul 25, 2024 · a07f121 · a07f121
2 parents e52a229 + a06262e
commit a07f121
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/popv/algorithms/_onclass.py b/popv/algorithms/_onclass.py
@@ -179,7 +179,7 @@ def predict(self, adata):
             pred_label = [train_model.i2co[ind] for ind in onclass_seen]
             pred_label_str = [clid_2_name[ind] for ind in pred_label]
             adata.obs[self.result_key] = pred_label_str
-            adata.obs["onclass_seen"] = pred_label_str
+            adata.obs[self.seen_result_key] = pred_label_str
         else:
             onclass_pred = train_model.Predict(corr_test_feature, use_normalize=False, refine=True, unseen_ratio=-1.0)
             pred_label = [train_model.i2co[ind] for ind in onclass_pred[2]]

diff --git a/tests/core/test_models.py b/tests/core/test_models.py
@@ -200,8 +200,7 @@ def test_annotation():
 
     adata = _get_test_anndata(mode='fast').adata
     popv.annotation.annotate_data(
-        adata, methods=["svm", "rf"],
-        save_path="tests/tmp_testing/popv_test_results/")
+        adata, save_path="tests/tmp_testing/popv_test_results/")
 
 
 def test_annotation_no_ontology():