From 1461a88e1db6026617797747095b2cee190f8e56 Mon Sep 17 00:00:00 2001
From: Hirokazu Ishida <hirokazu.ishida@sbintuitions.co.jp>
Date: Wed, 10 Sep 2025 15:01:10 +0900
Subject: [PATCH] maint(pyproject): use non-cuda jax if on macOS

---
 pyproject.toml | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index c4a06e532..cf74cce98 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -15,7 +15,8 @@ dependencies = [
     "fsspec[gcs]>=2024.6.0",
     "gym-aloha>=0.1.1",
     "imageio>=2.36.1",
-    "jax[cuda12]==0.5.3",
+    'jax[cuda12]==0.5.3; sys_platform == "linux"',
+    'jax==0.5.3; sys_platform == "darwin"',
     "jaxtyping==0.2.36",
     "lerobot",
     "ml_collections==1.0.0",