Merge pull request #1884 from JanSchulz/cat_fixes

BUG: Add compat code for pd.Categorical in pandas>=0.15
statsmodels · Aug 14, 2014 · 53dccdd · 53dccdd
2 parents 20c59d2 + 83be7a5
commit 53dccdd
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/statsmodels/tools/grouputils.py b/statsmodels/tools/grouputils.py
@@ -371,7 +371,11 @@ def labels(self):
         if hasattr(self.index, 'labels'):
             return self.index.labels
         else:  # pandas version issue here
-            return pd.Categorical(self.index).labels[None]
+            # Compat code for the labels -> codes change in pandas 0.15
+            # FIXME: use .codes directly when we don't want to support pandas < 0.15
+            tmp = pd.Categorical(self.index)
+            return getattr(tmp, "codes", tmp.labels)[None]
+
 
     @property
     def group_names(self):