Add in condition column check

ArcInstitute · Sep 17, 2024 · 6a47854 · 6a47854
1 parent 9094282
commit 6a47854
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/screenpro/phenoscore/__init__.py b/screenpro/phenoscore/__init__.py
@@ -56,6 +56,9 @@ def runPhenoScore(adata, cond_ref, cond_test, score_level, var_names='target', t
     """
     adat = adata.copy()
 
+    if 'condition' not in adat.obs.columns:
+        raise ValueError("The AnnData object must have a 'condition' column in its obs.")
+
     # format result name
     result_name = f'{cond_test}_vs_{cond_ref}'
     print(f'\t{cond_test} vs {cond_ref}')