Update weatherbench deps.

Before this, we allowed `numpy < 2`, but this was causing issues with time compatibility. Partial upgrades resulted in other issues, so I upgraded numpy/pandas/xarray to be inline with current Google versions.

PiperOrigin-RevId: 702104258

.github/workflows/ci-build.yml

@@ -16,8 +16,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        # TODO(#3): Support Python 3.10 and 3.11.
-        python-version: ["3.9"]
+        python-version: ["3.11"]
     steps:
     - name: Cancel previous
       uses: styfle/[email protected]

.github/workflows/lint.yml

@@ -16,8 +16,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        # TODO(#3): Support Python 3.10 and 3.11.
-        python-version: ["3.9"]
+        python-version: ["3.11"]
     steps:
     - name: Cancel previous
       uses: styfle/[email protected]

setup.py

@@ -17,12 +17,13 @@
 
 base_requires = [
     'apache_beam>=2.31.0',
+    'cftime>=1.6.2',
     'jax[cpu]',
-    'numpy',
-    'pandas==2.0.3',
+    'numpy>=2.1.3',
+    'pandas>=2.2.3',
     'scipy',
     'scikit-learn',
-    'xarray==2023.7.0',
+    'xarray>=2024.11.0',
     'xarray-beam',
     'zarr',
 ]
@@ -48,7 +49,7 @@
 
 setuptools.setup(
     name='weatherbench2',
-    version='0.2.0',
+    version='0.2.1',
     license='Apache 2.0',
     author='Google LLC',
     author_email='[email protected]',