get rid of fillna infer object warning

kp-forks · Mar 11, 2024 · 98a6c2d · 98a6c2d
1 parent c36c482
commit 98a6c2d
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/explainerdashboard/explainers.py b/explainerdashboard/explainers.py
@@ -761,9 +761,12 @@ def get_row_from_input(
 
         if len(inputs) == len(self.merged_cols):
             cols = self.columns_ranked_by_shap() if ranked_by_shap else self.merged_cols
-            df_merged = pd.DataFrame(dict(zip(cols, inputs)), index=[0]).fillna(
-                self.na_fill
-            )[self.merged_cols]
+            with pd.option_context("future.no_silent_downcasting", True):
+                df_merged = (
+                    pd.DataFrame(dict(zip(cols, inputs)), index=[0])
+                    .fillna(self.na_fill)
+                    .infer_objects(copy=False)[self.merged_cols]
+                )
             if return_merged:
                 return df_merged
             else: