Skip to content

Commit ab5dd2f

Browse files
authored
Scheduler type annotations (#9030)
* Reduce dask.order overhead by removing stripped_dep computation * Finish scheduler type annotations * fix tests
1 parent bda2d31 commit ab5dd2f

File tree

6 files changed

+131
-103
lines changed

6 files changed

+131
-103
lines changed

distributed/client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -886,7 +886,7 @@ def keys(self) -> Iterable[Key]:
886886
def _meta(self):
887887
return []
888888

889-
def _layer(self):
889+
def _layer(self) -> dict[Key, GraphNode]:
890890
dsk: _T_LowLevelGraph = {}
891891

892892
if not self.kwargs:

distributed/preloading.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -251,8 +251,8 @@ def __len__(self) -> int:
251251

252252
def process_preloads(
253253
dask_server: Server | Client,
254-
preload: str | list[str],
255-
preload_argv: list[str] | list[list[str]],
254+
preload: str | Sequence[str],
255+
preload_argv: Sequence[str] | Sequence[Sequence[str]],
256256
*,
257257
file_dir: str | None = None,
258258
) -> PreloadManager:

0 commit comments

Comments
 (0)