silence xarray pytype error in weatherbench2

Weatherbench authors · Weatherbench2 authors · commit 173d2e9527d7 · 2023-10-25T18:10:37.000-07:00
PiperOrigin-RevId: 576698744
diff --git a/weatherbench2/derived_variables.py b/weatherbench2/derived_variables.py
@@ -112,7 +112,7 @@ def _d_dx(field: xr.DataArray) -> xr.DataArray:
   cos_theta = np.cos(np.deg2rad(latitude))
   # TODO(shoyer): use a custom calculation with roll() instead of
   # differentiate() to calculate rolling over 360 to 0 degrees properly.
-  return _zero_poles(
+  return _zero_poles(  # pytype: disable=bad-return-type
       field.differentiate('longitude') / cos_theta / _METERS_PER_DEGREE
   )
 

Original file line number	Diff line number	Diff line change
`@@ -112,7 +112,7 @@ def _d_dx(field: xr.DataArray) -> xr.DataArray:`
`112`	`112`	`cos_theta = np.cos(np.deg2rad(latitude))`
`113`	`113`	`# TODO(shoyer): use a custom calculation with roll() instead of`
`114`	`114`	`# differentiate() to calculate rolling over 360 to 0 degrees properly.`
`115`		`- return _zero_poles(`
	`115`	`+ return _zero_poles( # pytype: disable=bad-return-type`
`116`	`116`	`field.differentiate('longitude') / cos_theta / _METERS_PER_DEGREE`
`117`	`117`	`)`
`118`	`118`