revert remove pip install jax == 0.3.22

datajoint · Apr 9, 2024 · cbf7ea5 · cbf7ea5
1 parent acec2e9
commit cbf7ea5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile
@@ -37,6 +37,9 @@ RUN \
     # clean up
     rm -rf /tmp/element-moseq/ && \
     apt-get clean
+
+# Install Keypoint-MoSeq (CPU version)
+RUN pip install "jax[cpu]==0.3.22" -f https://storage.googleapis.com/jax-releases/jax_releases.html
 
 ENV DJ_HOST fakeservices.datajoint.io
 ENV DJ_USER root