PR #41377: [CI/Build] Skip terratorch + torchgeo while PyPI has lightning quarantined

• Repository: vllm-project/vllm
• Author: stecasta
• State: closed | merged: True
• Link: https://github.com/vllm-project/vllm/pull/41377

Description (problem / solution / changelog)

Fixes #41376.

PyPI is currently serving the lightning project as quarantined: the simple index has <meta name="pypi:project-status" content="quarantined"> with zero release files exposed. Every published terratorch version transitively requires lightning, and torchgeo==0.7.0 (in requirements/test/rocm.in) requires lightning[pytorch-extra], so the pip-compile and pip-compile-rocm pre-commit hooks abort with:

× No solution found when resolving dependencies:
  ╰─▶ Because there are no versions of lightning and terratorch>=1.2.2 depends
      on lightning>=2.6.0, we can conclude that terratorch>=1.2.2 cannot be used.

--prerelease=allow does not help — the quarantine hides every version, including pre-releases.

This commit comments out terratorch in requirements/test/cuda.in + requirements/test/rocm.in, plus torchgeo==0.7.0 in rocm.in, and regenerates both lock files. Each comment block points at #41376 so the revert is mechanical once PyPI lifts the quarantine.

The Prithvi tests are skipped while this is in place. Production code that ships terratorch integration (vllm/model_executor/models/terratorch.py, vllm/renderers/registry.py) is unaffected — those are runtime-optional imports installed by users who actually want Prithvi.

Test plan

pre-commit run pip-compile      --files requirements/test/cuda.in
pre-commit run pip-compile-rocm --files requirements/test/rocm.in
pre-commit run pip-compile-xpu  --files requirements/test/xpu.in

Test result

All three hooks pass on the resulting tree:

pip-compile..............................................................Passed
pip-compile-rocm.........................................................Passed
pip-compile-xpu..........................................................Passed

grep -c "terratorch\|torchgeo" requirements/test/cuda.txt requirements/test/rocm.txt → 0 / 0 (cleanly removed from both lockfiles).

AI assistance was used in drafting this PR.

• The purpose of the PR, such as "Fix some issue (link existing issues this PR will resolve)".
• The test plan, such as providing test command.
• The test results, such as pasting the results comparison before and after, or e2e results.
• (Optional) The necessary documentation update, such as updating supported_models.md and examples for a new model.

Changed files

• requirements/test/cuda.in (modified, +5/-1)
• requirements/test/cuda.txt (modified, +11/-274)
• requirements/test/rocm.in (modified, +8/-2)
• requirements/test/rocm.txt (modified, +13/-265)
• tests/entrypoints/openai/chat_completion/test_vision_embeds.py (modified, +11/-0)
• tests/models/multimodal/pooling/test_prithvi_mae.py (modified, +5/-0)
• tests/models/test_terratorch.py (modified, +5/-0)
• tests/plugins_tests/test_terratorch_io_processor_plugins.py (modified, +5/-0)