Update weatherbench deps.

Before this, we allowed `numpy < 2`, and this now PiperOrigin-RevId: 702104258
google-research · Dec 3, 2024 · ed20fb6 · ed20fb6
1 parent 8308c9c
commit ed20fb6
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/setup.py b/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]',