From c12f8055b8777e942f97989c0f27a8223da7277e Mon Sep 17 00:00:00 2001 From: Bovard Doerschuk-Tiberi Date: Wed, 6 Nov 2024 08:01:54 -1000 Subject: [PATCH] sync gpu.Dockerfile with cpu.Dockerfile to fix tests (#298) --- docker/gpu.Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/gpu.Dockerfile b/docker/gpu.Dockerfile index d1258fc3..b0a977fe 100644 --- 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