unsloth/tests/vllm_compat/__init__.py
Daniel Han 9707b80520 CI(vllm): GRPO + fast_inference vLLM compat across 0.9 .. 0.15
Two new test files under tests/vllm_compat/, both CPU-only, both run
under tests/_zoo_aggressive_cuda_spoof.py so they pass on
ubuntu-latest without a GPU.

  test_unsloth_zoo_imports.py   import smoke for the 5 unsloth_zoo
                                modules the GRPO + fast_inference=True
                                path goes through. Strict assertions:
                                rl_replacements + empty_model MUST
                                import without pulling vllm
                                transitively (the use_vllm=False / no
                                fast_inference path on Colab without
                                vllm installed crashes if either of
                                them ever starts importing vllm).
                                vllm_utils + vllm_lora_request +
                                vllm_lora_worker_manager skip when
                                vllm is not on the runner; the symbol
                                test below covers them statically.

  test_vllm_pinned_symbols.py   parametrized across vLLM tags
                                v0.9.0, 0.9.2, 0.10.0, 0.10.2, 0.11.0,
                                0.12.0, 0.13.0, 0.14.0, 0.15.0. Each
                                cell fetches the relevant vllm source
                                files from github.com/vllm-project/vllm
                                at that tag (no pip install) and
                                asserts every symbol unsloth-zoo's
                                vllm_utils + vllm_lora_request +
                                vllm_lora_worker_manager hard-imports
                                or try/except imports is present.

Specifically catches:
  - vLLM PR #30253 split of vllm.lora.models -> {lora_model,
    model_manager}  (unsloth-zoo commit ec186187)
  - vLLM 0.14 gpu_model_runner.supports_tower_connector_lora call
    (unsloth-zoo commit e3072a23)
  - vLLM 0.15 LoRA manager kwarg rename (unsloth-zoo commit 2a80d543)
  - LoRARequest lora_path -> lora_dir rename progression
    (unsloth-zoo commits 888f79fd, e915bca1)
  - UNSLOTH_VLLM_STANDBY hard-error windows on vLLM 0.10.x and 0.14.x
    (unsloth-zoo commits 664e52ea, fa82dcc2) -- a sanity test asserts
    these guards stay in place.

Spoof contract: pynvml is sys.modules-stubbed at module top before
any unsloth_zoo import; torch.distributed is_available / is_initialized
are pinned to safe defaults via an autouse pytest fixture; the
existing _zoo_aggressive_cuda_spoof.apply() handles the
torch.cuda surface.

Validated locally: 51 passed in 7s.
2026-05-07 11:49:56 +00:00

0 lines
Python