From 59558ca129961d9bba965bfc95d3840c10366cf5 Mon Sep 17 00:00:00 2001 From: Bas Nijholt Date: Thu, 24 Oct 2024 12:47:28 -0700 Subject: [PATCH] . --- adaptive_scheduler/_executor.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/adaptive_scheduler/_executor.py b/adaptive_scheduler/_executor.py index 12afae33..18d20c59 100644 --- a/adaptive_scheduler/_executor.py +++ b/adaptive_scheduler/_executor.py @@ -94,15 +94,15 @@ def _get(self) -> Any | None: """Updates the state of the task and returns the result if the task is finished.""" i_learner, index = self.id_ learner, fname = self._learner_and_fname(load=False) + if self._state == "FINISHED": + return learner.data[index] + assert self.executor._run_manager is not None last_load_time = self.executor._run_manager._last_load_time.get(i_learner, 0) now = time.monotonic() time_since_last_load = now - last_load_time if time_since_last_load < self.min_load_interval: return None - if self._state == "FINISHED": - return learner.data[index] - try: mtime = os.path.getmtime(fname) # noqa: PTH204 except FileNotFoundError: @@ -130,7 +130,7 @@ def _learner_and_fname(self, *, load: bool = True) -> tuple[SequenceLearner, str i_learner, _ = self.id_ run_manager = self.executor._run_manager assert run_manager is not None - learner = run_manager.learners[i_learner] + learner: SequenceLearner = run_manager.learners[i_learner] # type: ignore[index] fname = run_manager.fnames[i_learner] if load and not learner.done(): learner.load(fname)