From f3e3a4c555ee3e0d240492a37a342e79d73b0ee1 Mon Sep 17 00:00:00 2001
From: Florian Pfaff <FlorianPfaff@users.noreply.github.com>
Date: Wed, 30 Oct 2024 11:03:59 +0100
Subject: [PATCH] Require jaxlib 0.4.33 or newer

---
 pyproject.toml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 5f09c425..f124db1c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,13 +20,13 @@ shapely = ">=2.0"
 [tool.poetry.extras]
 healpy_support = ["healpy"]
 pytorch_support = ["torch"]
-jax_support = ["jax", "jaxlib", "autograd"]
+jax_support = ["jax>=0.4.33", "jaxlib>=0.4.33", "autograd"]
 
 [tool.poetry.group.dev.dependencies]
 healpy = "*"
 torch = "*"
-jax = "*"
-jaxlib = "*"
+jax = ">=0.4.33"
+jaxlib = ">=0.4.33"
 autograd = "*"
 autopep8 = "^2.3.1"
 pytest = "*"