We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9bec841 commit 390e9a4Copy full SHA for 390e9a4
distributed/deploy/adaptive.py
@@ -20,8 +20,8 @@
20
if TYPE_CHECKING:
21
from typing_extensions import TypeAlias
22
23
+ import distributed
24
from distributed.deploy.cluster import Cluster
- from distributed.scheduler import WorkerState
25
26
logger = logging.getLogger(__name__)
27
@@ -113,7 +113,9 @@ def __init__(
113
maximum: int | float | None = None,
114
wait_count: int | None = None,
115
target_duration: str | float | timedelta | None = None,
116
- worker_key: Callable[[WorkerState], Hashable] | None = None,
+ worker_key: (
117
+ Callable[[distributed.scheduler.WorkerState], Hashable] | None
118
+ ) = None,
119
**kwargs: Any,
120
):
121
self.cluster = cluster
0 commit comments