[https://nvbugs/6075345][fix] test_llmapi_launch_multiple_tasks ignored the task_script parameter and always

[tensorrt-cicd]

Summary

• Fix for NVBugs 6075345: [TensorRT-LLM][L0][Pre-Merge][main] Test failed: unittest/llmapi/test_mpi_session.py
• Root cause: test_llmapi_launch_multiple_tasks ignored the task_script parameter and always ran _run_multi_llm_tasks.py, which fails with ModuleNotFoundError: No module named 'utils.llm_data' because the mpirun subprocess lacks the pytest sys.path setup needed for that import.
• Fix: Replace the hardcoded _run_multi_llm_tasks.py with the parametrized task_script variable so the test runs the intended MPI comm task scripts.
• 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/6075345

Summary by CodeRabbit

• Tests
  • Enhanced test parametrization to support multiple test scenarios with different configurations, improving test flexibility and coverage.

[coderabbitai]

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Pro Plus

Run ID: 9414d1a0-fe43-4e03-9376-15b036903c11

📥 Commits

Reviewing files that changed from the base of the PR and between ab315dd and ada760c.

📒 Files selected for processing (1)

• tests/unittest/llmapi/test_mpi_session.py

---

📝 Walkthrough

Walkthrough

A test method is modified to use a parametrized argument for selecting test scripts instead of hard-coding a single script name, enabling the same test to run with different script files under different parameterized test cases.

Changes

Cohort / File(s)	Summary
Test Parametrization tests/unittest/llmapi/test_mpi_session.py	Modified test_llmapi_launch_multiple_tasks to construct test_file path using the parametrized task_script argument instead of a hard-coded script name, allowing the test to execute different scripts based on parametrization values.

Estimated code review effort

🎯 1 (Trivial) | ⏱️ ~3 minutes

🚥 Pre-merge checks | ✅ 5

✅ Passed checks (5 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Title check	✅ Passed	The title clearly identifies the specific bug being fixed and the root cause (test ignoring the parametrized task_script parameter), directly matching the changeset.
Docstring Coverage	✅ Passed	Docstring coverage is 100.00% which is sufficient. The required threshold is 80.00%.
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.

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

✨ Finishing Touches

🧪 Generate unit tests (beta)

• Create PR with unit tests

Warning

Review ran into problems

🔥 Problems

Git: Failed to clone repository. Please run the @coderabbitai full review command to re-trigger a full review. If the issue persists, set path_filters to include or exclude specific files.

---

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

[Superjomn]

LGTM

[Superjomn]

/bot run

[tensorrt-cicd]

PR_Github #44926 [ run ] triggered by Bot. Commit: ada760c Link to invocation

[tensorrt-cicd]

PR_Github #44926 [ run ] completed with state SUCCESS. Commit: ada760c
/LLM/main/L0_MergeRequest_PR pipeline #35255 completed with status: 'FAILURE'

CI Report

⚠️ Action Required:

• Please check the failed tests and fix your PR
• If you cannot view the failures, ask the CI triggerer to share details
• Once fixed, request an NVIDIA team member to trigger CI again

Link to invocation

[Superjomn]

Close and continue the work on #13588