more reshape call into the more generic method

heal-research · Apr 4, 2024 · 7e485b4 · 7e485b4
1 parent 0c1e9c9
commit 7e485b4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyoperon/sklearn.py b/pyoperon/sklearn.py
@@ -579,7 +579,7 @@ def evaluate_model(self, model, X):
         X = check_array(X, accept_sparse=False)
         ds = op.Dataset(X)
         rg = op.Range(0, ds.Rows)
-        return op.Evaluate(model, ds, rg)
+        return op.Evaluate(model, ds, rg).reshape(-1,)
 
 
     def predict(self, X):
@@ -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,)
+        return self.evaluate_model(self.model_, X)