coerce Pandas categories into a column

rapidsai · Nov 30, 2022 · 710c6f6 · 710c6f6
1 parent c93caaa
commit 710c6f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cudf/cudf/core/column/column.py b/python/cudf/cudf/core/column/column.py
@@ -907,7 +907,7 @@ def as_categorical_column(self, dtype, **kwargs) -> ColumnBase:
             isinstance(dtype, pd.CategoricalDtype)
             and dtype.categories is not None
         ):
-            labels = self._label_encoding(cats=dtype.categories._column)
+            labels = self._label_encoding(cats=as_column(dtype.categories))
             if "ordered" in kwargs:
                 warnings.warn(
                     "Ignoring the `ordered` parameter passed in `**kwargs`, "