silence xarray pytype error in weatherbench2

PiperOrigin-RevId: 576698744
google-research · Oct 26, 2023 · 94705a3 · 94705a3
1 parent 338e411
commit 94705a3
Showing 1 changed file with 1 addition and 1 deletion.
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
   )