Correct category dtype check in ExperimentAxisQuery (#209)

single-cell-data · Aug 10, 2024 · 569c196 · 569c196
1 parent 5e3f5b4
commit 569c196
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python-spec/src/somacore/query/query.py b/python-spec/src/somacore/query/query.py
@@ -317,10 +317,10 @@ def to_anndata(
         # Drop unused categories on axis dataframes if requested
         if drop_levels:
             for name in ad.obs:
-                if isinstance(ad.obs[name], pd.CategoricalDtype):
+                if ad.obs[name].dtype.name == "category":
                     ad.obs[name] = ad.obs[name].cat.remove_unused_categories()
             for name in ad.var:
-                if isinstance(ad.var[name], pd.CategoricalDtype):
+                if ad.var[name].dtype.name == "category":
                     ad.var[name] = ad.var[name].cat.remove_unused_categories()
 
         return ad