Merge branch 'fix-cyclicboosting-rowcol' into cyc-boost-deprec

sktime · May 16, 2024 · 5969b70 · 5969b70
2 parents e9d7dae + 39d0e49
commit 5969b70
Showing 1 changed file with 5 additions and 5 deletions.
diff --git 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,