Add a torch.compile graph-break regression test for DiNTS

[nilapate]

Fixes #9159 .

Description

Follow-up to #9145, which fixed the DiNTS graph breaks from #9144 but added no test for that condition.

Adds TestDintsGraphBreaks to tests/networks/nets/test_dints_network.py: builds the grid from the issue, runs torch._dynamo.explain, asserts (graph_count, graph_break_count) == (1, 0)`.

Guarded by torch._dynamo.is_dynamo_supported() rather than a version check — (1, 0) verified on 2.8.0+cpu (the pyproject.toml floor). The test also raises the torch.__trace logger to WARNING for its duration (restored via addCleanup), since dynamo's DEBUG records are replayed by pytest on failure and bury the assertion message.

Types of changes

• Non-breaking change (fix or new feature that would not break existing functionality).
• Breaking change (fix or new feature that would cause existing functionality to change).
• New tests added to cover the changes.
• Integration tests passed locally by running ./runtests.sh -f -u --net --coverage.
• Quick tests passed locally by running ./runtests.sh --quick --unittests --disttests.
• In-line docstrings updated.
• Documentation updated, tested make html command in the docs/ folder.

[coderabbitai]

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

• Configuration used: Repository: Project-MONAI/MONAI/.coderabbit.yaml
• Review profile: CHILL
• Plan: Advanced
• Run ID: d7cfa624-ca84-4732-8c81-e2dfd85f84b0

📥 Commits

Reviewing files that changed from the base of the PR and between 6e358ab and 638b051.

📒 Files selected for processing (1)

• tests/networks/nets/test_dints_network.py

Included review availability: This review used your included allowance. Your plan provides up to 8 included reviews per hour; 7 remain after this review.

---

📝 Walkthrough

Walkthrough

Adds a Dynamo regression test for DiNTS. When Dynamo is supported, the test traces an evaluation-mode CPU model and asserts one graph and zero graph breaks. It suppresses trace logging during the test and restores the previous logger level afterward.

Priority: ⬇️ Low

Estimated code review effort: 2 (Simple) | ~8 minutes

Merge Risk: ⚪ Minimal · up to 638b0

The added test covers the reported DiNTS graph-break regression. No material merge-blocking risk is evident from the supplied change.

🚥 Pre-merge checks | ✅ 4 | ❌ 1

❌ Failed checks (1 warning)

Check name	Status	Explanation	Resolution
Docstring Coverage		Docstring coverage is 66.67% which is insufficient. The required threshold is 80.00%. Docstring coverage is scoped to functions touched by this diff. Analyzed 3 functions across 1 files.	Write docstrings for the functions missing them to satisfy the coverage threshold.

✅ Passed checks (4 passed)

Check name	Status	Explanation
Title check		The title clearly and concisely describes the main change: adding a torch.compile graph-break regression test for DiNTS.
Description check		The description includes the issue reference, change summary, test behavior, compatibility guard, logging cleanup, and change-type selections. It does not claim that the listed integration or quick te…
Linked Issues check		Issue #9159 requires a test in test_dints_network.py that asserts one graph and zero graph breaks, with a torch._dynamo.is_dynamo_supported() guard. The PR adds TestDintsGraphBreaks with `torch.…
Out of Scope Changes check		The change is limited to the requested regression test in tests/networks/nets/test_dints_network.py. The logging import, Dynamo usage, support guard, and logger restoration directly support the te…

• Fix all pre-merge checks with AI

✨ Finishing Touches

🧪 Generate unit tests (beta)

• Create a new PR

---

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.}