Merge pull request #50 from YosefLab/canergen-patch-1

Update _onclass.py to work in fast mode.
YosefLab · Jul 25, 2024 · a06262e · a06262e
2 parents 6a9a87c + 693220a
commit a06262e
Showing 1 changed file with 1 addition and 1 deletion.
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]]