MKV-000: Downcasting to float32

markovml · Dec 26, 2023 · a8e3b9b · a8e3b9b
1 parent 53cda9d
commit a8e3b9b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dataprep/eda/eda_frame.py b/dataprep/eda/eda_frame.py
@@ -120,7 +120,7 @@ def __init__(
             if isinstance(self._eda_dtypes[col], (Nominal, GeoGraphy)):
                 ddf[col] = ddf[col].apply(_to_str_if_not_na, meta=(col, "object"))
             elif isinstance(self._eda_dtypes[col], NumericString):
-                ddf[col] = ddf[col].astype(np.float).fillna(0.0)
+                ddf[col] = ddf[col].astype(np.float32).fillna(0.0)
             # transform pandas extension type to the numpy type,
             # to avoid computation issue of pandas type, e.g., #733.
             elif issubclass(type(ddf[col].dtype), pd.api.extensions.ExtensionDtype):