1000:1000

["-m" "nhx.rl.tasks.training" "--help"]

/opt/venv/bin/python

PATH=/opt/nemo_rl_venv/bin:/opt/nemo_rl_venv/bin:/usr/local/bin:/opt/cpython/bin:/usr/local/bin:/root/.local/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/mpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/amazon/efa/bin HOME=/home/ubuntu UV_CACHE_DIR=/home/ubuntu/.cache/uv

USERNAME=ubuntu USER_UID=1000 USER_GID=1000 NHX_COLLECT_SOURCES=0 /bin/bash -c python /tmp/remove-listed-files.py /tmp/file-removals.txt

USERNAME=ubuntu USER_UID=1000 USER_GID=1000 NHX_COLLECT_SOURCES=0 /bin/bash -c UV_BIN=uv-glue uv-glue run --no-project --python /opt/nemo_rl_venv/bin/python /usr/local/bin/collect-python-sdists.py --python /opt/nemo_rl_venv/bin/python --output /usr/share/nemo-helix/source-distributions/nhx-rl-training --baseline-freeze /tmp/base-python-freeze.txt --label platform-glue

NHX_COLLECT_SOURCES=0

NHX_COLLECT_SOURCES=0

RUN |3 USERNAME=ubuntu USER_UID=1000 USER_GID=1000 /bin/bash -c <<"EOF" bash -eu umask 022 export UV_CACHE_DIR=/root/.cache/uv for python in \
/opt/ray_venvs/nemo_rl.models.policy.workers.megatron_policy_worker.MegatronPolicyWorker/bin/python \
/opt/ray_venvs/nemo_rl.models.generation.vllm.vllm_worker.VllmGenerationWorker/bin/python \
/opt/ray_venvs/nemo_rl.models.generation.vllm.vllm_worker_async.VllmAsyncGenerationWorker/bin/python \
/opt/ray_venvs/nemo_rl.environments.nemo_gym.NemoGym/bin/python \
/opt/ray_venvs/nemo_rl.environments.sandbox.nemo_gym_actor.SandboxedGymActor/bin/python do uv-glue pip install --python "${python}" --no-cache --no-deps \
-e "/app/packages/sandboxed_gym[server,opensandbox,ray]" done EOF # buildkit

USERNAME=ubuntu USER_UID=1000 USER_GID=1000 /bin/bash -c umask 022 &&
  UV_CACHE_DIR=/root/.cache/uv uv-glue pip freeze --python /opt/nemo_rl_venv/bin/python --exclude-editable | grep -E '^[A-Za-z0-9._-]+==' | grep -Ev '^opentelemetry-' > /tmp/base-overrides.txt &&
  grep -Ev '^[[:space:]]*(#|$)' /tmp/opentelemetry-overrides.txt >> /tmp/base-overrides.txt &&
  UV_CACHE_DIR=/root/.cache/uv uv-glue pip install --python /opt/nemo_rl_venv/bin/python --no-cache --overrides /tmp/base-overrides.txt -e /app/packages/filesets -e /app/packages/nemo_helix_plugin -e /app/packages/nemo_helix_ext -e /app/packages/nhx_common -e /app/packages/nhx_customization_common -e "/app/packages/sandboxed_gym[server,ray]" -e "/app/services/rl[integrations]"