Skip to content

Commit 1094ed6

Browse files
committed
MAINT update function name
1 parent fcf3825 commit 1094ed6

File tree

3 files changed

+4
-7
lines changed

3 files changed

+4
-7
lines changed

autosklearn/automl.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -494,7 +494,7 @@ def send_warnings_to_log(message, category, filename, lineno,
494494

495495
random_state = np.random.RandomState(self._seed)
496496
for identifier in self.models_:
497-
if identifier in self.ensemble_.get_model_identifiers():
497+
if identifier in self.ensemble_.get_selected_model_identifiers():
498498
model = self.models_[identifier]
499499
# this updates the model inplace, it can then later be used in
500500
# predict method
@@ -554,7 +554,7 @@ def predict(self, X, batch_size=None, n_jobs=1):
554554
# Each process computes predictions in chunks of batch_size rows.
555555
all_predictions = joblib.Parallel(n_jobs=n_jobs)(
556556
joblib.delayed(_model_predict)(self, X, batch_size, identifier)
557-
for identifier in self.ensemble_.get_model_selected_identifiers())
557+
for identifier in self.ensemble_.get_selected_model_identifiers())
558558

559559
if len(all_predictions) == 0:
560560
raise ValueError('Something went wrong generating the predictions. '

autosklearn/ensembles/abstract_ensemble.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def get_models_with_weights(self, models):
6161

6262

6363
@abstractmethod
64-
def get_model_identifiers(self):
64+
def get_selected_model_identifiers(self):
6565
"""Return identifiers of models in the ensemble.
6666
6767
This includes models which have a weight of zero!

autosklearn/ensembles/ensemble_selection.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -222,10 +222,7 @@ def get_models_with_weights(self, models):
222222

223223
return output
224224

225-
def get_model_identifiers(self):
226-
return self.identifiers_
227-
228-
def get_model_selected_identifiers(self):
225+
def get_selected_model_identifiers(self):
229226
output = []
230227

231228
for i, weight in enumerate(self.weights_):

0 commit comments

Comments
 (0)