Merge pull request #85 from klarman-cell-observatory/yiming

Adjust filtering step for spatial data
lilab-bcb · Jan 11, 2022 · 5a243b3 · 5a243b3
2 parents 6c82c8c + 2c79d9b
commit 5a243b3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pegasusio/qc_utils.py b/pegasusio/qc_utils.py
@@ -162,7 +162,9 @@ def apply_qc_filters(unidata: UnimodalData, uns_white_list: str = None):
 
         unidata.obs.drop(columns=cols, inplace=True)
         if len(unidata.obsm) > 0:
-            unidata.obsm.clear()
+            for key in unidata.obsm.keys():
+                if key not in ['X_spatial']:
+                    del unidata.obsm[key]
         if len(unidata.varm) > 0:
             unidata.varm.clear()
         if uns_white_list is not None: