search

google-research / weatherbench2

A benchmark for the next generation of data-driven global weather models.
https://weatherbench2.readthedocs.io
Apache License 2.0
416 stars 42 forks source link

Misc. fixes for newer versions of Xarray #182

Open copybara-service[bot] opened 4 weeks ago

copybara-service[bot] commented 4 weeks ago

Misc. fixes for newer versions of Xarray

These fixes fall into three categories:

  1. Xarray's groupby no longer supports squeezing out size 1 dimensions. Instead, I've set squeeze=False and added explicit calls to .squeeze().
  2. Slight differences in dtypes/rounding (I believe these are indirectly due to changes required for NumPy 2.0 compatibility).
  3. Various cases where type checking is stricter (I've added type: ignore to these lines)