From c27eee49236212dca2b0032bfde157eacf1d4523 Mon Sep 17 00:00:00 2001 From: priyakasimbeg Date: Mon, 21 Oct 2024 13:17:24 -0700 Subject: [PATCH 1/2] Update setup.cfg Pin scipy version, without this some jax imports may break --- setup.cfg | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.cfg b/setup.cfg index 321020ad9..a5997e745 100644 --- a/setup.cfg +++ b/setup.cfg @@ -49,6 +49,7 @@ install_requires = clu==0.0.7 matplotlib>=3.7.2 tabulate==0.9.0 + scipy==1.11.4 python_requires = >=3.8 From cda6e46935ddbb6ddc475d40c231b9db8592f602 Mon Sep 17 00:00:00 2001 From: priyakasimbeg Date: Mon, 21 Oct 2024 23:49:27 +0000 Subject: [PATCH 2/2] move scipy dep --- setup.cfg | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/setup.cfg b/setup.cfg index a5997e745..eb570dafb 100644 --- a/setup.cfg +++ b/setup.cfg @@ -49,7 +49,6 @@ install_requires = clu==0.0.7 matplotlib>=3.7.2 tabulate==0.9.0 - scipy==1.11.4 python_requires = >=3.8 @@ -122,6 +121,8 @@ jax_core_deps = chex==0.1.7 ml_dtypes==0.2.0 protobuf==4.25.3 + scipy==1.11.4 + # JAX CPU jax_cpu =