sklearn.py: fix bug in the shape of the returned prediction

heal-research · Apr 4, 2024 · 5a41ebc · 5a41ebc
1 parent d0d0271
commit 5a41ebc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyoperon/sklearn.py b/pyoperon/sklearn.py
@@ -596,4 +596,4 @@ def predict(self, X):
             Returns an array of ones.
         """
         check_is_fitted(self)
-        return self.evaluate_model(self.model_, X).reshape(-1, 1)
+        return self.evaluate_model(self.model_, X).reshape(-1,)