[ROCM][CI] Fix AMD Examples Test Group

[Concurrensee]

This PR fixes AMD Examples Test Group

[mergify]

Documentation preview: https://vllm--30276.org.readthedocs.build/en/30276/

[gemini-code-assist]

Code Review

This pull request fixes a CI issue for AMD examples by setting the attention backend to FLEX_ATTENTION on ROCm platforms. While the use of os.environ is a valid approach, I've suggested a safer and more explicit alternative that modifies the parsed arguments locally. This avoids modifying global state and improves the code's clarity and robustness.

[mergify]

Hi @Concurrensee, the pre-commit checks have failed. Please run:

uv pip install pre-commit
pre-commit install
pre-commit run --all-files

Then, commit the changes and push to your branch.

For future commits, pre-commit will run automatically on changed files before each commit.

Tip

Is mypy or markdownlint failing?

mypy and markdownlint are run differently in CI. If the failure is related to either of these checks, please use the following commands to run them locally:

# For mypy (substitute "3.10" with the failing version if needed)
pre-commit run --hook-stage manual mypy-3.10
# For markdownlint
pre-commit run --hook-stage manual markdownlint

[tjtanaa]

Is this safe? will it override the vllm version?

I suggest following test.txt by specifying the packages directly, avoid using vllm[audio]==0.11.0