Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions vllm/executor/ray_distributed_executor.py
Original file line number Diff line number Diff line change
Expand Up @@ -559,6 +559,13 @@
envs.VLLM_USE_RAY_COMPILED_DAG_NCCL_CHANNEL)
logger.info("VLLM_USE_RAY_COMPILED_DAG_OVERLAP_COMM = %s",
envs.VLLM_USE_RAY_COMPILED_DAG_OVERLAP_COMM)
if os.environ.get("RAY_CGRAPH_get_timeout"): # noqa: SIM112
logger.info("RAY_CGRAPH_get_timeout = %s",
os.environ["RAY_CGRAPH_get_timeout"])
else:
os.environ["RAY_CGRAPH_get_timeout"] = "300" # noqa: SIM112

Check failure on line 566 in vllm/executor/ray_distributed_executor.py

View workflow job for this annotation

GitHub Actions / pre-commit

Ruff (SIM112)

vllm/executor/ray_distributed_executor.py:566:36: SIM112 Use capitalized environment variable `RAY_CGRAPH_GET_TIMEOUT` instead of `RAY_CGRAPH_get_timeout`
logger.info(
"RAY_CGRAPH_get_timeout is not set, setting to 300 seconds")
with InputNode() as input_data:
# Example DAG: PP=2, TP=4
#
Expand Down
Loading