ENH: Use context manager to suppress nan-categorical warning

quantopian · Jun 2, 2017 · 6b19e45 · 6b19e45
1 parent 69d1269
commit 6b19e45
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/zipline/utils/pandas_utils.py b/zipline/utils/pandas_utils.py
@@ -258,10 +258,11 @@ def categorical_df_concat(df_list, inplace=False):
         new_categories = sorted(
             set().union(
                 *(frame[col].cat.categories for frame in df_list)
-            ) - {None}
+            )
         )
 
-        for df in df_list:
-            df[col].cat.set_categories(new_categories, inplace=True)
+        with ignore_pandas_nan_categorical_warning():
+            for df in df_list:
+                df[col].cat.set_categories(new_categories, inplace=True)
 
     return pd.concat(df_list)