diff --git a/cu_cat/_table_vectorizer.py b/cu_cat/_table_vectorizer.py index fc24827b8..881713c71 100644 --- a/cu_cat/_table_vectorizer.py +++ b/cu_cat/_table_vectorizer.py @@ -541,7 +541,7 @@ def _auto_cast(self, X: pd.DataFrame) -> pd.DataFrame: num_col = X.select_dtypes(include=['int64','float64']).columns for i in num_col: X[i] = X[i].fillna(0) - X[i] = pd.to_numeric(X[i],downcast='float',errors='ignore') + X[i] = pd.to_numeric(X[i],downcast='float',errors='ignore') # type: ignore for col in X.columns: # Convert pandas' NaN value (pd.NA) to numpy NaN value (np.nan)