Merge branch 'chase/ray_install' of https://github.com/chaserileyrobe…

…rts/JAX-Toolbox into add-ray Co-authored-by: Chase Riley Roberts <chaser@nvidia.com>
NVIDIA · Sep 27, 2024 · 5581afa · 5581afa
2 parents 1a3febb + c23f238
commit 5581afa
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/.github/container/Dockerfile.base b/.github/container/Dockerfile.base
@@ -127,6 +127,12 @@ git commit -m 'Adds JAX_TOOLBOX_VCS_EQUIVALENCY as a trigger to treat all github
 EOF
 RUN pip install --upgrade --no-cache-dir -e /opt/pip pip-tools && rm -rf ~/.cache/*
 
+###############################################################################
+## Install Ray
+###############################################################################
+RUN echo "ray==2.37.0" >> /opt/pip-tools.d/requirements-ray.in
+EXPOSE 6379 8265 10001-10004
+
 ###############################################################################
 ## Install TCPx
 ###############################################################################