feat(evaulator): Add dedicated Gym task image for agent evaluation - #1384
Merged
JashG merged 4 commits intoAug 19, 2026
Merged
Conversation
Signed-off-by: Jash Gulabrai <jgulabrai@nvidia.com>
JashG
force-pushed
the
jgulabrai/AALGO-489-gym-base-image
branch
from
August 19, 2026 14:19
17e8691 to
bf0e0bf
Compare
JashG
marked this pull request as ready for review
August 19, 2026 14:19
Signed-off-by: Jash Gulabrai <jgulabrai@nvidia.com>
Signed-off-by: Jash Gulabrai <jgulabrai@nvidia.com>
…pulls image Signed-off-by: Jash Gulabrai <jgulabrai@nvidia.com>
JashG
merged commit Aug 19, 2026
eaa30a3
into
jgulabrai/AALGO-494-gym-integration-tests
4 checks passed
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
NOTE: Target branch is set to #1315 as this is a follow-up to that PR.
Gym agent evaluations currently depend on a CI-specific variant of the shared CPU task image. This PR replaces it with a dedicated, releasable
nmp-gym-tasksimage that isolates Gym and Ray dependencies from the sharednmp-cpu-tasksimage.Changes
Gym task image
nmp-gym-tasksimage based onnmp-cpu-tasks./opt/gym-venv.Evaluator routing
GymRunnerTargetevaluations tonmp-gym-tasks.nmp-cpu-tasks.NEMO_EVALUATOR_GYM_TASKS_IMAGEfor overriding the fully qualified Gym image.Image publication
nmp-gym-tasksin the standard CPU image build group.CI and validation
tiktoken.nmp-cpu-tasks-gym-e2eimage and tag-alias workflow.Dependency maintenance
Common Patterns with Customizer Images
Customizer also generates training images with Gym as a dependency. This PR follows the same relevant patterns:
runtimestage from the cache-onlysmoke-teststage.docker-bake.hcl.Type of Change
Quality Gates
Verification
Signed-off-by:traileruv run pre-commit run -apasses, or any blocked checks are identified belowTargeted validation:
nmp-gym-tasksimage:1 passed in 177.20s.docker buildx bake nmp-gym-tasks-smoke-testpassed and importednemo-gym 0.5.0,ray 2.57.0, andtiktoken 0.13.0.docker buildx bake --print docker-cpu docker-cpu-ciconfirmed the runtime and smoke targets are wired into the expected build groups.14 passed.nmp-gym-tasks.