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: 703143594

.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]

docs/contributing.md

@@ -33,12 +33,9 @@ for this purpose. If possible, consider reviewing [Google's Code Review Guide](h
 
 ### Development Setup
 
-After cloning the project, we recommend creating a local Python3.9 environment
+After cloning the project, we recommend creating a local Python3.11 environment
 for development (try [Miniconda](https://docs.conda.io/en/latest/miniconda.html)).
 
-> Note: We'll soon support Python 3.10 & 3.11 ([#3](https://github.com/google-research/weatherbench2/issues/3)).
-> At that point, we will recommend creating a Python3.10 environment.
-
 To locally install the project for development, please run:
 ```shell
 pip install -e ".[tests]"

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]',