diff --git a/skpro/regression/cyclic_boosting.py b/skpro/regression/cyclic_boosting.py index a6c719ddc..ad4bf836a 100644 --- a/skpro/regression/cyclic_boosting.py +++ b/skpro/regression/cyclic_boosting.py @@ -346,11 +346,11 @@ def _predict_proba(self, X): # Johnson Quantile-Parameterized Distributions params = { "alpha": self.alpha, - "qv_low": self.quantile_values[0], - "qv_median": self.quantile_values[1], - "qv_high": self.quantile_values[2], - "lower": self._lower, - "upper": self._upper, + "qv_low": self.quantile_values[0].reshape(-1, 1), + "qv_median": self.quantile_values[1].reshape(-1, 1), + "qv_high": self.quantile_values[2].reshape(-1, 1), + "lower": self.lower, + "upper": self.upper, "version": self.dist_type, "dist_shape": self.dist_shape, "index": index,