diff --git a/autoPyTorch/api/base_task.py b/autoPyTorch/api/base_task.py index be214ac51..e50862948 100644 --- a/autoPyTorch/api/base_task.py +++ b/autoPyTorch/api/base_task.py @@ -686,7 +686,6 @@ def _load_best_individual_model(self) -> SingleBest: backend=self._backend, ) - return ensemble def _cleanup(self) -> None: @@ -1344,7 +1343,7 @@ def _search( self.ensemble_.identifiers_ ) ) - + self._cleanup() return self