From efba19458cfafd3f147d0ad596fb8a8d16c293b0 Mon Sep 17 00:00:00 2001
From: Chong Shen Ng <chong.shen@flower.ai>
Date: Wed, 6 Nov 2024 21:41:13 +0000
Subject: [PATCH] Rollback e2e-jax

---
 e2e/e2e-jax/pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/e2e/e2e-jax/pyproject.toml b/e2e/e2e-jax/pyproject.toml
index 65d55a243884..b259f66a7bc3 100644
--- a/e2e/e2e-jax/pyproject.toml
+++ b/e2e/e2e-jax/pyproject.toml
@@ -12,7 +12,7 @@ dependencies = [
     "jax==0.4.13",
     "jaxlib==0.4.13",
     "scikit-learn>=1.1.1,<2.0.0",
-    "numpy>=2.0.0",
+    "numpy>=1.21.4,<2.0.0",
 ]
 
 [tool.hatch.build.targets.wheel]