Bug fix for Netflix#1023

valayDave · Apr 28, 2022 · 45c5a16 · 45c5a16
1 parent 169ac15
commit 45c5a16
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/metaflow/plugins/cards/card_modules/convert_to_native_type.py b/metaflow/plugins/cards/card_modules/convert_to_native_type.py
@@ -313,6 +313,7 @@ def _parse_pandas_dataframe(self, data_object, truncate=True):
             if data[col].dtype == "datetime64[ns]":
                 data[col] = data[col].dt.strftime(time_format)
 
+        data = data.astype(object).where(data.notnull(), None)
         data_vals = data.values.tolist()
         for row, idx in zip(data_vals, index_column.values.tolist()):
             row.insert(0, idx)