No public description

PiperOrigin-RevId: 584373729
google-research · Nov 21, 2023 · a613ca7 · a613ca7
1 parent 545481f
commit a613ca7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/resample_daily.py b/scripts/resample_daily.py
@@ -257,7 +257,7 @@ def _is_not_duplicated(kv: tuple[xbeam.Key, xr.Dataset], stat: str) -> bool:
             obs, input_chunks, split_vars=True, num_threads=NUM_THREADS.value
         )
         | 'RechunkIn'
-        >> xbeam.Rechunk(
+        >> xbeam.Rechunk(  # pytype: disable=wrong-arg-types
             obs.sizes, input_chunks, in_working_chunks, itemsize=itemsize
         )
     )
@@ -287,7 +287,7 @@ def _is_not_duplicated(kv: tuple[xbeam.Key, xr.Dataset], stat: str) -> bool:
         pcolls
         | beam.Flatten()
         | 'RechunkOut'
-        >> xbeam.Rechunk(
+        >> xbeam.Rechunk(  # pytype: disable=wrong-arg-types
             rsmp_template.sizes,
             out_working_chunks,
             output_chunks,