Merge pull request #110 from Quantmetry/angoho_fix_simple_methods

Fix: the problem of a column including nans in test but training set

Author: JulienRoussel77

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)