Refactored seed_dict assignment logic (#77)

* Updated version to 0.1.19 * Refactored seed_dict assignment logic * Update test cases
INCATools · Jul 2, 2024 · 43c40eb · 43c40eb
1 parent 848ca96
commit 43c40eb
Show file tree

Hide file tree

Showing 5 changed files with 122 additions and 101 deletions.
diff --git a/pandasaurus_cxg/anndata_enricher.py b/pandasaurus_cxg/anndata_enricher.py
@@ -48,7 +48,7 @@ def __init__(
         self.anndata = anndata
         self.seed_dict = dict(
             self.anndata.obs.drop_duplicates(subset=[cell_type_field, "cell_type"])
-            .dropna()[[cell_type_field, "cell_type"]]
+            .dropna(subset=[cell_type_field, "cell_type"])[[cell_type_field, "cell_type"]]
             .values
         )
         # "unknown" patch