Skip to content

fix: AUT-2228 route partial reruns to the current attempt - #4048

Merged
chtruong814 merged 1 commit into
NVIDIA-NeMo:mainfrom
svcnemo-autobot:fix/partial-rerun-runner-labels
Sep 8, 2026
Merged

fix: AUT-2228 route partial reruns to the current attempt#4048
chtruong814 merged 1 commit into
NVIDIA-NeMo:mainfrom
svcnemo-autobot:fix/partial-rerun-runner-labels

Conversation

@svcnemo-autobot

Copy link
Copy Markdown
Contributor

Background

Partial reruns must target runners allocated for the current workflow attempt.

What changed

Updated the ephemeral router to v1.9.2, passed its attempt-independent base through organization-member preflight, and reconstructed attempt-qualified labels at 9 scheduling boundaries. Static and GB200 labels remain unchanged.

Details

flowchart LR
  R[Router base] --> P[Member preflight]
  P --> J[Current-attempt jobs]
Loading

Legacy labels remain unchanged.

Tested

Passed YAML parsing, routing/output wiring and boundary-count checks, whitespace, size, conflict-marker, and private-key scans. Full pre-commit could not start because the typed checkout does not expose Git metadata; CI remains authoritative.

Signed-off-by: svcnemo-autobot <svcnemo-autobot@nvidia.com>
@svcnemo-autobot
svcnemo-autobot requested a review from a team as a code owner September 8, 2026 17:32
@copy-pr-bot

copy-pr-bot Bot commented Sep 8, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@github-actions github-actions Bot added the CI Relating to CI label Sep 8, 2026
@chtruong814 chtruong814 added the CI:docs Run doctest label Sep 8, 2026
@chtruong814
chtruong814 enabled auto-merge (squash) September 8, 2026 17:41
@svcnvidia-nemo-ci svcnvidia-nemo-ci added CI:L1 Run doctests, unit tests, and functional tests and removed CI:docs Run doctest labels Sep 8, 2026
@svcnvidia-nemo-ci

Copy link
Copy Markdown
Contributor

/ok to test 136e53c

@chtruong814
chtruong814 merged commit aa7eed3 into NVIDIA-NeMo:main Sep 8, 2026
186 of 188 checks passed
rrs45 added a commit that referenced this pull request Sep 8, 2026
Picks up periodic rollout checkpointing (#3924) and the partial-rerun
routing fix (#4048), which landed while the previous merge was being
verified.

One conflict, in the critic training block: main now raises
_optimizer_commit_in_progress before the value_training timer so a
periodic snapshot cannot land mid-update. Kept, with our
rl.sc.value_training span re-applied around the timer and the flag left
where main set it -- outside the span, since it guards the whole
irreversible update rather than the measured region.

Signed-off-by: Raj Singh <rajsin@nvidia.com>
asolergi-nv added a commit that referenced this pull request Sep 9, 2026
Two conflicts, both pure adjacency -- independent additions landing at the same
file position, so both sides are kept:

- nemo_rl/algorithms/single_controller.py: this branch's
  _SUPERVISOR_DRAIN_TIMEOUT_S constant sits where #3924 (periodic rollout
  checkpointing) added _MAX_CONSECUTIVE_ROLLOUT_CHECKPOINT_FAILURES and the
  _RolloutCheckpointCut dataclass. Neither refers to the other.

- tests/unit/environments/test_nemo_gym_utils.py: #4014 (L2 Gym-to-RL rollout
  acceptance) appended a spinup-cleanup test to the same end-of-file this
  branch appended TestUnresolvedAgentRefsAreDiagnosable to. main's function
  first, this branch's class after it.

The other three commits (#3730 colocated MInf, #4002 actor venv list from a
Python leaf module, #4048 partial rerun routing) merged clean.

Verified after the merge: the restart keys are still in both SC exemplars,
on_dead_shard is still absent from the config and both yamls, and 1287 tests
pass across single_controller, fleet_health, weight_sync and the gym utils.

Signed-off-by: asolergibert <asolergibert@nvidia.com>
Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CI:L1 Run doctests, unit tests, and functional tests CI Relating to CI

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants