narrow condition for Float64 sanitization in Altair (#247)

lux-org · Feb 20, 2021 · c4229a0 · c4229a0
1 parent dfa64eb
commit c4229a0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lux/vislib/altair/AltairChart.py b/lux/vislib/altair/AltairChart.py
@@ -118,7 +118,8 @@ def initialize_chart(self):
     @classmethod
     def sanitize_dataframe(self, df):
         for attr in df.columns:
-            if pd.api.types.is_float_dtype(df[attr].dtype):  # Check if dtype is unrecognized by Altair
+            # Check if dtype is unrecognized by Altair (#247)
+            if str(df[attr].dtype) == "Float64":
                 df[attr] = df[attr].astype(np.float64)
 
             # Altair can not visualize non-string columns