No public description

Weatherbench authors · Weatherbench2 authors · commit 059311adfba9 · 2023-11-21T10:54:25.000-08:00
PiperOrigin-RevId: 584243967
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,

Original file line number	Diff line number	Diff line change
`@@ -257,7 +257,7 @@ def _is_not_duplicated(kv: tuple[xbeam.Key, xr.Dataset], stat: str) -> bool:`
`257`	`257`	`obs, input_chunks, split_vars=True, num_threads=NUM_THREADS.value`
`258`	`258`	`)`
`259`	`259`	`\| 'RechunkIn'`
`260`		`- >> xbeam.Rechunk(`
	`260`	`+ >> xbeam.Rechunk( # pytype: disable=wrong-arg-types`
`261`	`261`	`obs.sizes, input_chunks, in_working_chunks, itemsize=itemsize`
`262`	`262`	`)`
`263`	`263`	`)`
`@@ -287,7 +287,7 @@ def _is_not_duplicated(kv: tuple[xbeam.Key, xr.Dataset], stat: str) -> bool:`
`287`	`287`	`pcolls`
`288`	`288`	`\| beam.Flatten()`
`289`	`289`	`\| 'RechunkOut'`
`290`		`- >> xbeam.Rechunk(`
	`290`	`+ >> xbeam.Rechunk( # pytype: disable=wrong-arg-types`
`291`	`291`	`rsmp_template.sizes,`
`292`	`292`	`out_working_chunks,`
`293`	`293`	`output_chunks,`