check ifattr exists

single-cell-data · Sep 11, 2023 · 29b6982 · 29b6982
1 parent 99f2049
commit 29b6982
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/apis/python/src/tiledbsoma/_arrow_types.py b/apis/python/src/tiledbsoma/_arrow_types.py
@@ -193,11 +193,14 @@ def df_to_arrow(df: pd.DataFrame) -> pa.Table:
     for key in df:
         column = df[key]
         if is_categorical_dtype(column.dtype):
-            column = column.astype("category")
-            df[key] = pd.core.arrays.categorical.Categorical(
-                values=column,
-                categories=column.categories,
-                ordered=bool(column.ordered),
+            if hasattr(column.values, "categories"):
+                categories = column.values.categories
+
+            if hasattr(column.values, "ordered"):
+                ordered = column.values.ordered
+
+            df[key] = pd.Categorical(
+                values=column, categories=categories, ordered=ordered
             )
 
     arrow_table = pa.Table.from_pandas(df)