From ad52a76c96fb4b0f8b1b935c063a6a27d413a35e Mon Sep 17 00:00:00 2001 From: Rebecca Chen Date: Thu, 29 Feb 2024 14:44:04 -0800 Subject: [PATCH] Silence some pytype errors. PiperOrigin-RevId: 611601569 --- xarray_beam/_src/core.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/xarray_beam/_src/core.py b/xarray_beam/_src/core.py index 43fed40..7130f34 100644 --- a/xarray_beam/_src/core.py +++ b/xarray_beam/_src/core.py @@ -306,7 +306,7 @@ def __init__( raise ValueError( f"chunks key {dim!r} is not a dimension on the provided dataset(s)" ) - expanded_chunks = normalize_expanded_chunks(chunks, self._first.sizes) + expanded_chunks = normalize_expanded_chunks(chunks, self._first.sizes) # pytype: disable=wrong-arg-types # always-use-property-annotation self.expanded_chunks = expanded_chunks self.split_vars = split_vars self.num_threads = num_threads @@ -395,7 +395,7 @@ def _iter_all_keys(self) -> Iterator[Key]: relevant_offsets = { k: v for k, v in self.offsets.items() if k in variable.dims } - yield from iter_chunk_keys(relevant_offsets, vars={name}) + yield from iter_chunk_keys(relevant_offsets, vars={name}) # pytype: disable=wrong-arg-types # always-use-property-annotation def _iter_shard_keys( self, shard_id: Optional[int], var_name: Optional[str] @@ -429,7 +429,7 @@ def _shard_inputs(self) -> List[Tuple[Optional[int], Optional[str]]]: inputs.extend([(i, name) for i in range(self.shard_count)]) else: inputs.append((None, name)) - return inputs + return inputs # pytype: disable=bad-return-type # always-use-property-annotation def _key_to_chunks(self, key: Key) -> Iterator[Tuple[Key, DatasetOrDatasets]]: """Convert a Key into an in-memory (Key, xarray.Dataset) pair."""