detail

skrub-data · Dec 1, 2023 · 3131039 · 3131039
1 parent a132d73
commit 3131039
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/skrub/_interpolation_joiner.py b/skrub/_interpolation_joiner.py
@@ -299,7 +299,9 @@ def transform(self, X):
         )
         prediction_results = self._check_prediction_results(prediction_results)
         predictions = [res["predictions"] for res in prediction_results]
-        predictions = _multi_add_column_name_suffix(predictions, self.suffix)
+        predictions = [
+            _join_utils.add_column_name_suffix(df, self.suffix) for df in predictions
+        ]
         for part in predictions:
             part.index = main_table.index
         return pd.concat([main_table] + predictions, axis=1)
@@ -430,7 +432,3 @@ def _predict(key_values, columns, estimator, propagate_exceptions):
         "columns": columns,
         "shape": (key_values.shape[0], len(columns)),
     }
-
-
-def _multi_add_column_name_suffix(dataframes, suffix):
-    return [_join_utils.add_column_name_suffix(df, suffix) for df in dataframes]