From c8ee2a25118fe6136d2720a3e3cc0a9d40f36b74 Mon Sep 17 00:00:00 2001
From: Eugene Rakhmatulin <eugr@eugr.com>
Date: Thu, 26 Mar 2026 18:15:09 -0700
Subject: [PATCH] Perform node count check in any mode

---
 launch-cluster.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/launch-cluster.sh b/launch-cluster.sh
index 9ae1088..b20603e 100755
--- a/launch-cluster.sh
+++ b/launch-cluster.sh
@@ -991,8 +991,8 @@ exec_no_ray_cluster() {
 }
 
 if [[ "$ACTION" == "exec" ]]; then
-    # For --no-ray, trim (or error on) PEER_NODES based on declared parallelism
-    if [[ "$NO_RAY_MODE" == "true" ]]; then
+    # Trim (or error on) PEER_NODES based on declared parallelism, for any multi-node exec
+    if [[ "$SOLO_MODE" != "true" && ${#PEER_NODES[@]} -gt 0 ]]; then
         if [[ "$LAUNCH_SCRIPT_MODE" == "true" ]]; then
             cmd_text=$(cat "$LAUNCH_SCRIPT_PATH" 2>/dev/null || true)
         else