diff --git a/skpro/regression/cyclic_boosting.py b/skpro/regression/cyclic_boosting.py index 19620db78..230f0c5f7 100644 --- a/skpro/regression/cyclic_boosting.py +++ b/skpro/regression/cyclic_boosting.py @@ -284,9 +284,9 @@ 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], + "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.version,