From 199d2b549289290f86283be42ae47fca0c217edc Mon Sep 17 00:00:00 2001
From: Anushan Fernando <anushanf@google.com>
Date: Wed, 11 Sep 2024 10:02:32 -0700
Subject: [PATCH] Exclude netcdf4 version 1.7.1 due to buggy behaviour. See
 discussion on https://github.com/Unidata/netcdf4-python/issues/1343.

Additionally because of not using 1.7.1 netcdf (which adds support for np>=2 compatibility https://docs.xarray.dev/en/stable/whats-new.html#id10) restrict xarray versions from very latest versions.

PiperOrigin-RevId: 673435320
---
 pyproject.toml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 027c3839..d27d0ed4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,8 +29,8 @@ dependencies = [
     "chex>=0.1.85",
     "equinox @ git+https://github.com/patrick-kidger/equinox@1e601672d38d2c4d483535070a3572d8e8508a20",
     "PyYAML>=6.0.1",
-    "xarray>=2023.12.0",
-    "netcdf4>=1.6.5",
+    "xarray>=2023.12.0,<=2024.06.0",
+    "netcdf4>=1.6.5,<1.7.1",
     "h5netcdf>=1.3.0",
     "scipy>=1.12.0",
     "jaxtyping>=0.2.28",