Fix pandas FutureWarning (#711)

Quantco · Oct 16, 2023 · e7ef80a · e7ef80a
1 parent 75cc52c
commit e7ef80a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/glum/_glm.py b/src/glum/_glm.py
@@ -2255,7 +2255,7 @@ def _set_up_and_check_fit_args(
                         _name_categorical_variables(
                             dtype.categories, column, self.drop_first
                         )
-                        if pd.api.types.is_categorical_dtype(dtype)
+                        if isinstance(dtype, pd.CategoricalDtype)
                         else [column]
                         for column, dtype in zip(X.columns, X.dtypes)
                     )