FIX predict

rushter · May 26, 2016 · 430216a · 430216a
1 parent 206f1a1
commit 430216a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/heamy/estimator.py b/heamy/estimator.py
@@ -105,7 +105,8 @@ def _predict(self, X_train, y_train, X_test, y_test=None):
                 if result.shape[1] == 2:
                     result = result[:, 1]
         else:
-            result = self._estimator(X_train, y_train, X_test, y_test=None, **self.parameters)
+            result = self._estimator(X_train=X_train, y_train=y_train,
+                                     X_test=X_test, y_test=y_test, **self.parameters)
 
         return result