patch: Multi output support for TrainOnlyTransformerMixin (#700)

koaning · Sep 9, 2024 · 28f2600 · 28f2600
1 parent 236f491
commit 28f2600
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sklego/common.py b/sklego/common.py
@@ -81,7 +81,7 @@ def fit(self, X, y=None):
         if y is None:
             check_array(X, estimator=self)
         else:
-            check_X_y(X, y, estimator=self)
+            check_X_y(X, y, estimator=self, multi_output=True)
         self.X_hash_ = self._hash(X)
         self.n_features_in_ = X.shape[1]
         return self