[https://nvbugs/6550727][fix] Pin _force_non_greedy_for_capture=False on the test double so it models a…

[trtllm-agent]

Summary

• Root cause: The test passed a bare Mock() as spec_metadata, and Mock auto-vivifies attributes, so the capture-override getattr(..., False) returned a truthy child Mock and tripped the assert.
• Fix: Pin _force_non_greedy_for_capture=False on the test double so it models a real SpecMetadata outside warmup, leaving the production serving-correctness guard intact.
• Automated fix generated by repair-bot

Test plan

• Verify fix on the same GPU type as the original failure
• Check for regressions in related tests

Links

• Bug: https://nvbugs/6550727

Dev Engineer Review

• The mock now sets _force_non_greedy_for_capture=False.
• This matches SpecMetadata behavior outside warmup.
• The production serving-correctness guard remains unchanged.
• No API, configuration, or performance changes were found.

QA Engineer Review

• Modified the speculative-overlap test in tests/unittest/_torch/executor/test_pytorch_model_engine.py.
• No integration test-list files were modified.
• Coverage in tests/integration/test_lists/ could not be confirmed.

Verdict: needs follow-up

[coderabbitai]

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: a151e604-2ce5-412d-afb1-b9b9c17fb463

📥 Commits

Reviewing files that changed from the base of the PR and between d8b25c5 and 9a651fc.

📒 Files selected for processing (1)

• tests/unittest/_torch/executor/test_pytorch_model_engine.py

---

Walkthrough

The speculative-overlap test now explicitly sets _force_non_greedy_for_capture to False on its spec_metadata mock.

Changes

Capture Mock Behavior

Layer / File(s)	Summary
Explicit capture flag configuration tests/unittest/_torch/executor/test_pytorch_model_engine.py	The test replaces the bare Mock with a mock that explicitly disables _force_non_greedy_for_capture.

Estimated code review effort: 1 (Trivial) | ~2 minutes

Suggested reviewers: bowenfu

🚥 Pre-merge checks | ✅ 5

✅ Passed checks (5 passed)

Check name	Status	Explanation
Title check	✅ Passed	The title identifies the NVBugs issue, fix type, and primary change to the test double.
Description check	✅ Passed	The description explains the root cause, fix, test plan, and related bug link, but uses different headings from the template.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.

✨ Finishing Touches

🧪 Generate unit tests (beta)

• Create PR with unit tests

---

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