pin scipy version

dattalab · Apr 17, 2024 · 8b73813 · 8b73813
1 parent 79fd392
commit 8b73813
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/docs/source/install.rst b/docs/source/install.rst
@@ -69,7 +69,7 @@ Install jax using one of the lines below::
    # MacOS or Linux (CPU)
    pip install "jax[cpu]==0.3.22" -f https://storage.googleapis.com/jax-releases/jax_releases.html
 
-   # MacOS or Linux (GPU)
+   # MacOS or Linux (GPU with CUDA 11.X)
    pip install "jax[cuda11_cudnn82]==0.3.22" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 
    # Windows (CPU)

diff --git a/setup.cfg b/setup.cfg
@@ -35,6 +35,7 @@ install_requires =
     commentjson
     jaxtyping==0.2.14
     etils==1.5.2
+    scipy==1.11.3
     jax-moseq==0.2.2
 
 [options.package_data]