[Testing] Set VLLM_WORKER_MULTIPROC_METHOD for e2e testing (#1569)

dsikka · web-flow · commit a144d8b44c55 · 2025-06-18T19:35:50.000Z
SUMMARY: - Set `VLLM_WORKER_MULTIPROC_METHOD` fo e2e vllm testing, as per vllm-project/vllm#14535 (comment) - With that, our e2e is green: https://github.com/neuralmagic/llm-compressor-testing/actions/runs/15739077015/job/44359835739
diff --git a/tests/e2e/vLLM/test_vllm.py b/tests/e2e/vLLM/test_vllm.py
@@ -30,7 +30,7 @@
 )
 SKIP_HF_UPLOAD = os.environ.get("SKIP_HF_UPLOAD", "")
 TIMINGS_DIR = os.environ.get("TIMINGS_DIR", "timings/e2e-test_vllm")
-
+os.environ["VLLM_WORKER_MULTIPROC_METHOD"] = "spawn"
 EXPECTED_SAVED_FILES = [
     "config.json",
     r"^model(?:-\d{5}-of-\d{5})?\.safetensors$",

Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@`
`30`	`30`	`)`
`31`	`31`	`SKIP_HF_UPLOAD = os.environ.get("SKIP_HF_UPLOAD", "")`
`32`	`32`	`TIMINGS_DIR = os.environ.get("TIMINGS_DIR", "timings/e2e-test_vllm")`
`33`		`-`
	`33`	`+os.environ["VLLM_WORKER_MULTIPROC_METHOD"] = "spawn"`
`34`	`34`	`EXPECTED_SAVED_FILES = [`
`35`	`35`	`"config.json",`
`36`	`36`	`r"^model(?:-\d{5}-of-\d{5})?\.safetensors$",`