Merge pull request #413 from Kaggle/consolidate_installs

Consolidate python installs

Author: SohierDane

docker/cpu.Dockerfile

@@ -30,9 +30,7 @@ ADD ./README.md ./README.md
 ADD ./MANIFEST.in ./MANIFEST.in
 ADD ./kaggle_environments ./kaggle_environments
 
-
-# install kaggle-environments. vec-noise cannot be installed with uv's more stringent checks.
-RUN pip install vec-noise && uv pip install accelerate jax gymnax==0.0.8 litellm && uv pip install . && pytest
+RUN uv pip install . && pytest
 
 # SET UP KAGGLE-ENVIRONMENTS CHESS
 # minimal package to reduce memory footprint

docker/gpu.Dockerfile

@@ -29,7 +29,7 @@ ADD ./pyproject.toml ./pyproject.toml
 ADD ./README.md ./README.md
 ADD ./MANIFEST.in ./MANIFEST.in
 ADD ./kaggle_environments ./kaggle_environments
-# vec-noise cannot be installed with uv's more stringent checks.
-RUN pip install vec-noise && uv pip install Flask bitsandbytes accelerate jax gymnax==0.0.8 && uv pip install . && pytest
+
+RUN uv pip install . && pytest
 
 CMD kaggle-environments

docker/open_spiel.Dockerfile

@@ -35,7 +35,7 @@ ADD ./MANIFEST.in ./MANIFEST.in
 ADD ./kaggle_environments ./kaggle_environments
 
 # install kaggle-environments
-RUN pip install Flask bitsandbytes accelerate vec-noise jax gymnax==0.0.8 && pip install . && pytest
+RUN uv pip install . && pytest
 
 # begin OpenSpiel integration
 

pyproject.toml

@@ -6,10 +6,15 @@ authors = [
     {name = "Kaggle", email = "[email protected]"},
 ]
 dependencies = [
+    "accelerate",
+    "bitsandbytes",
     "Chessnut >= 0.4.1",
     "Flask >= 1.1.2",
     "gymnasium == 1.2.0",
+    "gymnax==0.0.8",
+    "jax",
     "jsonschema >= 3.0.1",
+    "litellm",
     "numpy >= 2.2.6",
     "open_spiel >= 1.6.0",
     "pettingzoo == 1.24.0",