fix: test set has a column including nans, but this column does not contain nans in train set. Simple methodes can not impute this column.

Author: anhkhoangoho

qolmat/imputations/imputers.py

@@ -159,10 +159,9 @@ def fit(self, X: pd.DataFrame, y=None) -> Self:
         else:
             self.ngroups_ = pd.Series(0, index=df.index).rename("_ngroup")
 
-        cols_with_nans = df.columns[df.isna().any()]
         self._setup_fit()
         if self.columnwise:
-            for col in cols_with_nans:
+            for col in df.columns:
                 self._fit_allgroups(df[[col]], col=col)
         else:
             self._fit_allgroups(df)