diff --git a/xarray_beam/_src/combiners.py b/xarray_beam/_src/combiners.py index 1cf3e89..0e1adc7 100644 --- a/xarray_beam/_src/combiners.py +++ b/xarray_beam/_src/combiners.py @@ -288,7 +288,7 @@ def _optimal_fanout_bins( ) # The dict of candidates is empty if chunks_count=1, in which can there's no # need to use a combiner. - return min(candidates, key=candidates.get) if candidates else () + return min(candidates, key=candidates.get) if candidates else () # pyrefly: ignore[no-matching-overload] @dataclasses.dataclass diff --git a/xarray_beam/_src/dataset.py b/xarray_beam/_src/dataset.py index 396819e..c43d246 100644 --- a/xarray_beam/_src/dataset.py +++ b/xarray_beam/_src/dataset.py @@ -159,7 +159,7 @@ def normalize_chunks( if ... in chunks: # pyrefly: ignore[not-iterable, unsupported-operation] default_chunks = chunks[...] # pyrefly: ignore[bad-index] - chunks = {k: chunks.get(k, default_chunks) for k in template.dims} # pyrefly: ignore[bad-assignment] + chunks = {k: chunks.get(k, default_chunks) for k in template.dims} # pyrefly: ignore[bad-assignment, missing-attribute] defaults = previous_chunks if previous_chunks else template.sizes chunks: dict[str, int | str] = {**defaults, **chunks} # pytype: disable=annotation-type-mismatch diff --git a/xarray_beam/_src/zarr.py b/xarray_beam/_src/zarr.py index 61713f4..1758192 100644 --- a/xarray_beam/_src/zarr.py +++ b/xarray_beam/_src/zarr.py @@ -125,7 +125,7 @@ def make_template( """ if lazy_vars is None: lazy_vars = set(dataset.keys()) # pyrefly: ignore[bad-assignment] - lazy_vars.update(k for k in dataset.coords if k not in dataset.indexes) + lazy_vars.update(k for k in dataset.coords if k not in dataset.indexes) # pyrefly: ignore[missing-attribute] result = dataset.copy()