Update _onclass.py to work in fast mode.

YosefLab · Jul 25, 2024 · 693220a · 693220a
1 parent e52a229
commit 693220a
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]]