Skip to content

Commit 390e9a4

Browse files
authored
Fix doc build (#8892)
1 parent 9bec841 commit 390e9a4

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

distributed/deploy/adaptive.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@
2020
if TYPE_CHECKING:
2121
from typing_extensions import TypeAlias
2222

23+
import distributed
2324
from distributed.deploy.cluster import Cluster
24-
from distributed.scheduler import WorkerState
2525

2626
logger = logging.getLogger(__name__)
2727

@@ -113,7 +113,9 @@ def __init__(
113113
maximum: int | float | None = None,
114114
wait_count: int | None = None,
115115
target_duration: str | float | timedelta | None = None,
116-
worker_key: Callable[[WorkerState], Hashable] | None = None,
116+
worker_key: (
117+
Callable[[distributed.scheduler.WorkerState], Hashable] | None
118+
) = None,
117119
**kwargs: Any,
118120
):
119121
self.cluster = cluster

0 commit comments

Comments
 (0)