sync gpu.Dockerfile with cpu.Dockerfile to fix tests (#298)

Kaggle · Nov 6, 2024 · c12f805 · c12f805
1 parent 8b80549
commit c12f805
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/gpu.Dockerfile b/docker/gpu.Dockerfile
@@ -29,6 +29,6 @@ ADD ./setup.py ./setup.py
 ADD ./README.md ./README.md
 ADD ./MANIFEST.in ./MANIFEST.in
 ADD ./kaggle_environments ./kaggle_environments
-RUN pip install Flask bitsandbytes accelerate vec-noise && pip install . && pytest
+RUN pip install Flask bitsandbytes accelerate vec-noise jax gymnax==0.0.8 && pip install . && pytest
 
 CMD kaggle-environments