diff --git a/sktime/forecasting/compose/_ensemble.py b/sktime/forecasting/compose/_ensemble.py index f523479d63f..e4211319b48 100644 --- a/sktime/forecasting/compose/_ensemble.py +++ b/sktime/forecasting/compose/_ensemble.py @@ -364,8 +364,14 @@ def _predict(self, fh, X): """ names, _ = self._check_forecasters() y_pred = pd.concat(self._predict_forecasters(fh, X), axis=1, keys=names) - y_pred = y_pred.groupby(level=1, axis=1).agg( - _aggregate, self.aggfunc, self.weights + y_pred = ( + y_pred.T.groupby(level=1) + .agg( + lambda y, aggfunc, weights: _aggregate(y.T, aggfunc, weights), + self.aggfunc, + self.weights, + ) + .T ) return y_pred