From 82e83f9ef1edaa6d1c182cfc831a275130403950 Mon Sep 17 00:00:00 2001
From: ppdebreuck <pierre-paul.debreuck@uclouvain.be>
Date: Sat, 13 Nov 2021 16:40:50 +0100
Subject: [PATCH] fix evaluate

---
 modnet/models/vanilla.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modnet/models/vanilla.py b/modnet/models/vanilla.py
index 552c3d52..a725eb6d 100644
--- a/modnet/models/vanilla.py
+++ b/modnet/models/vanilla.py
@@ -654,7 +654,7 @@ def evaluate(self, test_data: MODData) -> pd.DataFrame:
             x = self._scaler.transform(x)
             x = np.nan_to_num(x, nan=-1)
 
-        y_pred = self.model.predict(x)
+        y_pred = np.array(self.model.predict(x))
         if len(y_pred.shape) == 2:
             y_pred = np.array([y_pred])
         score = []