diff --git a/autoPyTorch/api/base_task.py b/autoPyTorch/api/base_task.py index 3504668d8..0cbfbcda5 100644 --- a/autoPyTorch/api/base_task.py +++ b/autoPyTorch/api/base_task.py @@ -634,7 +634,6 @@ def _load_best_individual_model(self) -> SingleBest: backend=self._backend, ) - return ensemble def _cleanup(self) -> None: @@ -1277,7 +1276,7 @@ def _search( self.ensemble_.identifiers_ ) ) - + self._cleanup() return self