FIX estimator signature for regression

automl · Feb 24, 2017 · 2e29eba · 2e29eba
1 parent f5608e2
commit 2e29eba
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/autosklearn/estimators.py b/autosklearn/estimators.py
@@ -434,7 +434,9 @@ def fit(self, X, y,
         """
         # Fit is supposed to be idempotent!
         # But not if we use share_mode.
-        return super(AutoSklearnRegressor, self).fit(X, y, metric, feat_type, dataset_name)
+        return super(AutoSklearnRegressor, self).fit(X=X, y=y, metric=metric,
+                                                     feat_type=feat_type,
+                                                     dataset_name=dataset_name)
 
     def predict(self, X):
         """Predict regression target for X.
@@ -559,5 +561,4 @@ def fit(self, X, y,
             dataset_name=None,
             ):
         return self._automl.fit(X=X, y=y, task=REGRESSION, metric=metric,
-                                loss=loss, feat_type=feat_type,
-                                dataset_name=dataset_name)
+                                feat_type=feat_type, dataset_name=dataset_name)