From ed20fb64e7d4ea17dfb927cd1f4dd159232467af Mon Sep 17 00:00:00 2001
From: Ian Langmore <langmore@google.com>
Date: Mon, 2 Dec 2024 15:30:12 -0800
Subject: [PATCH] Update weatherbench deps.

Before this, we allowed `numpy < 2`, and this now

PiperOrigin-RevId: 702104258
---
 setup.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/setup.py b/setup.py
index 4f1b984..618399c 100644
--- 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='noreply@google.com',