FIX problem with extra dimension that makes linear interpolation failing

update not working file Fix in Linear class
mathLab · Nov 15, 2023 · 063013c · 063013c
1 parent e57fe51
commit 063013c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ezyrb/linear.py b/ezyrb/linear.py
@@ -55,4 +55,4 @@ def predict(self, new_point):
         :return: the interpolated values.
         :rtype: numpy.ndarray
         """
-        return self.interpolator(new_point)
+        return self.interpolator(new_point).squeeze()