build(sdk): temporarily pin jax in tests because of breaking changes (#…

…4632)
wandb · Dec 14, 2022 · e086c26 · e086c26
1 parent 39d50c3
commit e086c26
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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