From b74687154d774678a4c98aa36b887772832aeec1 Mon Sep 17 00:00:00 2001
From: LocalCascadeEnsemble <kfauvel.lce@gmail.com>
Date: Sun, 11 Sep 2022 21:18:47 +0200
Subject: [PATCH] -

---
 lce/_lcetree.py | 2 +-
 lce/_version.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/lce/_lcetree.py b/lce/_lcetree.py
index 133f986..8a9e281 100644
--- a/lce/_lcetree.py
+++ b/lce/_lcetree.py
@@ -1210,7 +1210,7 @@ def _predict(node, X, y_pred_final=None):
         X = np.concatenate((index, X), axis=1)
         y_pred = _predict(self.tree, X, None)
         y_pred = y_pred[y_pred[:, 0].argsort()]
-        y_pred = np.squeeze(y_pred[:, 1:])
+        y_pred = y_pred[:, 1:]
         return y_pred
 
     def set_params(self, **params):
diff --git a/lce/_version.py b/lce/_version.py
index 75cf783..493f741 100644
--- a/lce/_version.py
+++ b/lce/_version.py
@@ -1 +1 @@
-__version__ = "0.2.9"
+__version__ = "0.3.0"