diff --git a/parsl/dataflow/job_error_handler.py b/parsl/dataflow/job_error_handler.py index 44dfdc0f96..2acf97f9fd 100644 --- a/parsl/dataflow/job_error_handler.py +++ b/parsl/dataflow/job_error_handler.py @@ -1,16 +1,19 @@ +from __future__ import annotations + from typing import List, Dict -from parsl.dataflow.job_status_poller import PollItem +import parsl.dataflow.job_status_poller as jsp +# from parsl.dataflow.job_status_poller import PollItem from parsl.executors.base import ParslExecutor from parsl.providers.base import JobStatus, JobState class JobErrorHandler: - def run(self, status: List[PollItem]): + def run(self, status: List[jsp.PollItem]): for es in status: self._check_irrecoverable_executor(es) - def _check_irrecoverable_executor(self, es: PollItem): + def _check_irrecoverable_executor(self, es: jsp.PollItem): if not es.executor.error_management_enabled: return es.executor.handle_errors(self, es.status) diff --git a/parsl/dataflow/strategy.py b/parsl/dataflow/strategy.py index 23ae8132f9..e92704c91f 100644 --- a/parsl/dataflow/strategy.py +++ b/parsl/dataflow/strategy.py @@ -1,10 +1,12 @@ +from __future__ import annotations import logging import time import math import warnings from typing import Dict, List, Optional -from parsl.dataflow.job_status_poller import PollItem +# from parsl.dataflow.job_status_poller import PollItem +import parsl.dataflow.job_status_poller as jsp from parsl.executors import HighThroughputExecutor from parsl.executors.base import ParslExecutor from parsl.executors.status_handling import BlockProviderExecutor @@ -134,7 +136,7 @@ def add_executors(self, executors): for executor in executors: self.executors[executor.label] = {'idle_since': None} - def _strategy_noop(self, status: List[PollItem]) -> None: + def _strategy_noop(self, status: List[jsp.PollItem]) -> None: """Do nothing. """ logger.debug("strategy_noop: doing nothing")