From 8f825100b3d19104336c9863670345268077fb5d Mon Sep 17 00:00:00 2001 From: Xarray-Beam authors Date: Wed, 5 Aug 2026 12:39:21 -0700 Subject: [PATCH] Add pyrefly suppressions PiperOrigin-RevId: 959811586 --- xarray_beam/_src/combiners.py | 2 +- xarray_beam/_src/dataset.py | 2 +- xarray_beam/_src/zarr.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/xarray_beam/_src/combiners.py b/xarray_beam/_src/combiners.py index 0e1adc7..1cf3e89 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 () # pyrefly: ignore[no-matching-overload] + return min(candidates, key=candidates.get) if candidates else () @dataclasses.dataclass diff --git a/xarray_beam/_src/dataset.py b/xarray_beam/_src/dataset.py index c43d246..396819e 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, missing-attribute] + chunks = {k: chunks.get(k, default_chunks) for k in template.dims} # pyrefly: ignore[bad-assignment] 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 1758192..61713f4 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) # pyrefly: ignore[missing-attribute] + lazy_vars.update(k for k in dataset.coords if k not in dataset.indexes) result = dataset.copy()