revert: Downgrade NVRx / Upgrade nemo-run

[chtruong814]

What does this PR do ?

revert: Downgrade NVRx / Upgrade nemo-run

The nvrx test has been failing since that got meged. Reverting for now until we can fix it better.

#2623

Changelog

• Add specific line by line info of high level changes in this PR.

GitHub Actions CI

See the CI sectionin the Contributing doc for how to trigger the CI. A Nvidia developer will need to approve and trigger the CI for external contributors.

Before your PR is "Ready for review"

Pre checks:

• Make sure you read and followed Contributor guidelines
• Did you write any new necessary tests?
• Did you add or update any necessary documentation?
• Does the PR affect components that are optional to install? (Ex: Numba, Pynini, Apex etc)
  • Reviewer: Does the PR have correct import guards for all optional libraries?

If you haven't finished some of the above items you can still open "Draft" PR.

Additional Information

• Related to # (issue)

Summary by CodeRabbit

• Tests
  • Added test markers to improve test organization and management.

[coderabbitai]

📝 Walkthrough

Walkthrough

A pytest marker decorator @pytest.mark.pleasefixme is added to the test_nvrx_straggler_detection_end_to_end function in the test file. The function signature and test logic remain unchanged. This annotation marks the test for future attention without modifying its behavior.

Changes

Cohort / File(s)	Summary
Test Marker Addition tests/functional_tests/training/test_nvrx_straggler.py	Added @pytest.mark.pleasefixme decorator to test_nvrx_straggler_detection_end_to_end function to flag the test for later review or fixes.

Estimated code review effort

🎯 1 (Trivial) | ⏱️ ~2 minutes

Suggested reviewers

• yaoyu-33
• thomasdhc

🚥 Pre-merge checks | ✅ 3 | ❌ 1

❌ Failed checks (1 warning)

Check name	Status	Explanation	Resolution
Title check	⚠️ Warning	The PR title 'revert: Downgrade NVRx / Upgrade nemo-run' does not match the actual changes, which only add a pytest marker to skip a test. The objectives indicate the intention is to skip the nvrx straggler test, not to downgrade/upgrade packages.	Update the PR title to accurately reflect the change, such as 'ci: Skip nvrx straggler test until fixed' which aligns with the stated objectives.

✅ Passed checks (3 passed)

Check name	Status	Explanation
Docstring Coverage	✅ Passed	Docstring coverage is 100.00% which is sufficient. The required threshold is 80.00%.
Test Results For Major Changes	✅ Passed	PR adds only a single test skip decorator with no impact on functionality, performance, or behavior—qualifying as a minor change requiring no test documentation.
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.

_{✏️ Tip: You can configure your own custom pre-merge checks in the settings.}

✨ Finishing Touches

• 📝 Generate docstrings (stacked PR)
• 📝 Generate docstrings (commit on current branch)

🧪 Generate unit tests (beta)

• Create PR with unit tests
• Post copyable unit tests in a comment
• Commit unit tests in branch chtruong/skip-nvrx

Tip

Try Coding Plans. Let us write the prompt for your AI agent so you can ship faster (with fewer bugs).
Share your feedback on Discord.

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands and usage tips.}

[coderabbitai]

Actionable comments posted: 1

🤖 Prompt for all review comments with AI agents

Verify each finding against the current code and only fix it if needed.

Inline comments:
In `@tests/functional_tests/training/test_nvrx_straggler.py`:
- Line 241: The test file uses the decorator `@pytest.mark.pleasefixme` but never
imports pytest, causing an F821; add an import statement for pytest (e.g.,
import pytest) near the other test imports so the symbol pytest used by the
decorator is defined and the test module can be imported; update any import
grouping in the file to keep imports organized around the existing test helpers
and fixtures.

---

ℹ️ Review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Pro

Run ID: 55988e11-47b5-4f55-b3b2-aeb578c9b3e6

📥 Commits

Reviewing files that changed from the base of the PR and between b730ab6 and 60892f2.

📒 Files selected for processing (1)

• tests/functional_tests/training/test_nvrx_straggler.py

[coderabbitai]

⚠️ Potential issue | 🔴 Critical

Add missing pytest import for the new decorator

Line 241 introduces @pytest.mark.pleasefixme, but pytest is not imported, which triggers F821 and breaks CI/module import.

✅ Proposed fix

 import argparse
 import logging
 import os
 import tempfile
 import time
 
+import pytest
 import torch

📝 Committable suggestion

‼️ IMPORTANT
Carefully review the code before committing. Ensure that it accurately replaces the highlighted code, contains no missing lines, and has no issues with indentation. Thoroughly test & benchmark the code to ensure it meets the requirements.

Suggested change

	@pytest.mark.pleasefixme
	import argparse
	import logging
	import os
	import tempfile
	import time
	import pytest
	import torch

🧰 Tools

🪛 Flake8 (7.3.0)

[error] 241-241: undefined name 'pytest'

(F821)

🪛 GitHub Actions: CICD NeMo

[error] 241-241: ruff: F821 Undefined name 'pytest'.

---

[warning] 241-241: ruff: Selection E266 has no effect because preview is not enabled.

🪛 Ruff (0.15.2)

[error] 241-241: Undefined name pytest

(F821)

🤖 Prompt for AI Agents

Verify each finding against the current code and only fix it if needed.

In `@tests/functional_tests/training/test_nvrx_straggler.py` at line 241, The test
file uses the decorator `@pytest.mark.pleasefixme` but never imports pytest,
causing an F821; add an import statement for pytest (e.g., import pytest) near
the other test imports so the symbol pytest used by the decorator is defined and
the test module can be imported; update any import grouping in the file to keep
imports organized around the existing test helpers and fixtures.

[ananthsub]

i think it was #2623

[chtruong814]

Let me try reverting that

[chtruong814]

Merging now to unblock main branch