diff --git a/autoPyTorch/api/base_task.py b/autoPyTorch/api/base_task.py index bf713a82b..56781c9ab 100644 --- a/autoPyTorch/api/base_task.py +++ b/autoPyTorch/api/base_task.py @@ -618,7 +618,6 @@ def _load_best_individual_model(self) -> SingleBest: backend=self._backend, ) - return ensemble def _cleanup(self) -> None: @@ -1261,7 +1260,7 @@ def _search( self.ensemble_.identifiers_ ) ) - + self._cleanup() return self