We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 3e739c2 + a8f0552 commit afac1aaCopy full SHA for afac1aa
statsmodels/__init__.py
@@ -1,7 +1,10 @@
1
+from statsmodels.compat.patsy import monkey_patch_cat_dtype
2
+
3
from statsmodels._version import __version__, __version_tuple__
4
5
__version_info__ = __version_tuple__
6
7
+monkey_patch_cat_dtype()
8
9
debug_warnings = False
10
statsmodels/compat/patsy.py
@@ -0,0 +1,16 @@
+from statsmodels.compat.pandas import PD_LT_2
+import pandas as pd
+import patsy.util
+def _safe_is_pandas_categorical_dtype(dt):
+ if PD_LT_2:
+ return pd.api.types.is_categorical_dtype(dt)
+ return isinstance(dt, pd.CategoricalDtype)
11
12
13
+def monkey_patch_cat_dtype():
14
+ patsy.util.safe_is_pandas_categorical_dtype = (
15
+ _safe_is_pandas_categorical_dtype
16
+ )
0 commit comments