Merge branch 'dev' into patch_em_mle

Author: vm-aifluence-jro

qolmat/benchmark/missing_patterns.py

@@ -662,7 +662,7 @@ def fit(self, X: pd.DataFrame) -> GroupedHoleGenerator:
 
         super().fit(X)
 
-        if self.n_splits > len(np.unique(self.groups_num)):
+        if self.n_splits > self.ngroups.nunique():
             raise ValueError("n_samples has to be smaller than the number of groups.")
 
         return self

qolmat/imputations/models.py

@@ -540,7 +540,6 @@ def fit_transform(self, df: pd.DataFrame) -> pd.DataFrame:
             imputed = pd.DataFrame()
             cols_with_nans = df.columns[df.isna().any()]
             for col in cols_with_nans:
-                print(col)
                 imputed_signal, _, _ = self.rpca.fit_transform(signal=df[col].values)
                 imputed[col] = imputed_signal
         imputed.index = df.index