software-engineering/spark-vllm-docker
2
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

Handle new copy hosts setup in run-recipe.py

Browse Source
This commit is contained in:
Eugene Rakhmatulin
2026-03-26 16:45:35 -07:00
parent f163ca69de
commit 00c16746e5
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
run-recipe.py
View File

@@ -954,6 +954,7 @@ Examples:
discovered_env = run_autodiscover() discovered_env = run_autodiscover()
if discovered_env and discovered_env.get("CLUSTER_NODES"): if discovered_env and discovered_env.get("CLUSTER_NODES"):
env = discovered_env # use freshly loaded env from autodiscover
nodes = parse_nodes(discovered_env["CLUSTER_NODES"]) nodes = parse_nodes(discovered_env["CLUSTER_NODES"])
nodes_from_env = True nodes_from_env = True
@@ -1000,8 +1001,17 @@ Examples:
print(f" 2. Remove nodes from .env: {sys.argv[0]} --show-env") print(f" 2. Remove nodes from .env: {sys.argv[0]} --show-env")
return 1 return 1
# Determine copy targets for cluster deployments # Determine copy targets for build/model distribution.
copy_targets = worker_nodes if is_cluster else None # Prefer COPY_HOSTS from .env (may differ from CLUSTER_NODES in mesh mode),
# fall back to worker_nodes derived from CLUSTER_NODES.
if is_cluster:
copy_hosts_str = env.get("COPY_HOSTS")
if copy_hosts_str:
copy_targets = [h.strip() for h in copy_hosts_str.split(",") if h.strip()]
else:
copy_targets = worker_nodes
else:
copy_targets = None
if args.dry_run: if args.dry_run:
print("=== Dry Run ===") print("=== Dry Run ===")