Skip to content

Commit

Permalink
Add more type annotations to dask builder
Browse files Browse the repository at this point in the history
  • Loading branch information
Kirill888 committed May 23, 2024
1 parent 6c4287e commit 713e6f2
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions odc/loader/_builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -208,9 +208,9 @@ def __call__(

for block_idx in np.ndindex(shape_in_blocks):
ti, yi, xi = block_idx[0], block_idx[ydim], block_idx[ydim + 1]
srcs = []
srcs_keys: list[list[tuple[str, int]]] = []
for _ti in tchunk_range[ti]:
srcs.append(
srcs_keys.append(
[
(src_key, src_idx)
for src_idx in self.tyx_bins.get((_ti, yi, xi), [])
Expand All @@ -220,7 +220,7 @@ def __call__(

dsk[(band_key, *block_idx)] = (
_dask_loader_tyx,
srcs,
srcs_keys,
gbt_dask_key,
quote((yi, xi)),
quote(prefix_dims),
Expand Down

0 comments on commit 713e6f2

Please sign in to comment.