Skip to content

Fix docker build and add training dependencies (kubernetes, r2e-gym)#1703

Open
niting wants to merge 1 commit into
google:mainfrom
niting:fix-docker-build
Open

Fix docker build and add training dependencies (kubernetes, r2e-gym)#1703
niting wants to merge 1 commit into
google:mainfrom
niting:fix-docker-build

Conversation

@niting

@niting niting commented Jul 17, 2026

Copy link
Copy Markdown
Contributor

Add relevant changes to get DeepSWE training to work on the Tunix stack. This includes dependency additions and fixes to import failures on examples.deepswe

@niting niting changed the title Fix docker build by installing curl and uv via pip Fix docker build and add training dependencies (kubernetes, r2e-gym) Jul 17, 2026
@niting
niting force-pushed the fix-docker-build branch from 38ac6e2 to 675fd34 Compare July 17, 2026 21:55
- Fixes the Docker build by installing curl and uv via pip.
- Installs kubernetes and r2e-gym at the end of the Dockerfile.
- Fixes examples.deepswe import error by removing tunix_root from sys.path in train_deepswe_nb.py to avoid shadowing by internal tunix/examples package.
Comment thread Dockerfile

RUN bash /app/scripts/install_tunix_vllm_requirement.sh
RUN pip install kubernetes
RUN pip install --no-deps git+https://github.com/r2e-gym/r2e-gym.git@0d94c4eb9431cd195c55a7ea3abd54006c9a1735

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Let's make it optional instead of default

Comment thread Dockerfile
# Install system dependencies, including Python 3 and pip
RUN apt-get update && \
apt-get install -y build-essential git python3 python3-pip && \
apt-get install -y build-essential git python3 python3-pip curl && \

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

If you replace the curl install of uv with pip, do you still need curl here?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants