fix predict_proba for classifer

Author: mabryj2

autosklearn/automl.py

@@ -943,7 +943,7 @@ def predict(self, X, batch_size=None, n_jobs=1):
             return predicted_classes
 
     def predict_proba(self, X, batch_size=None, n_jobs=1):
-        return self._automl.predict(X, batch_size=batch_size, n_jobs=n_jobs)
+        return super().predict(X, batch_size=batch_size, n_jobs=n_jobs)
 
 
 class AutoMLRegressor(BaseAutoML):

autosklearn/estimators.py

@@ -308,6 +308,10 @@ def refit(self, X, y):
     def predict(self, X, batch_size=None, n_jobs=1):
         return self._automl.predict(X, batch_size=batch_size, n_jobs=n_jobs)
 
+    def predict_proba(self, X, batch_size=None, n_jobs=1):
+        return self._automl.predict_proba(
+             X, batch_size=batch_size, n_jobs=n_jobs)
+
     def score(self, X, y):
         return self._automl.score(X, y)
 
@@ -428,7 +432,7 @@ def predict_proba(self, X, batch_size=None, n_jobs=1):
             The predicted class probabilities.
 
         """
-        return self._automl.predict_proba(
+        return super().predict_proba(
             X, batch_size=batch_size, n_jobs=n_jobs)