From e086c2668598fdb928b19b57b4891d8c87a59d3d Mon Sep 17 00:00:00 2001
From: Jeff Raubitschek <jeff@wandb.com>
Date: Wed, 14 Dec 2022 06:03:54 -0800
Subject: [PATCH] build(sdk): temporarily pin jax in tests because of breaking
 changes (#4632)

---
 requirements_dev.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements_dev.txt b/requirements_dev.txt
index 047f20a4cc4..dcbfe8f6dad 100644
--- a/requirements_dev.txt
+++ b/requirements_dev.txt
@@ -27,7 +27,7 @@ docker
 stable_baselines3
 tensorboard
 gym
-jax[cpu]; sys_platform == 'darwin' or sys_platform == 'linux'
+jax[cpu]<0.4; sys_platform == 'darwin' or sys_platform == 'linux'
 fastcore; python_version > '3.6'
 fastcore==1.3.29; python_version == '3.6'
 pyarrow