Fix command check

Signed-off-by: 林联辉 <[email protected]>
kubeflow · Oct 31, 2024 · 9dc8b00 · 9dc8b00
1 parent f5066f1
commit 9dc8b00
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/argsbuilder/serving_distributed.go b/pkg/argsbuilder/serving_distributed.go
@@ -153,6 +153,10 @@ func (s *DistributedServingArgsBuilder) check() error {
 		if s.args.MasterCommand != "" || s.args.WorkerCommand != "" {
 			return fmt.Errorf("--command and --master-command/--worker-command can not be set at the same time")
 		}
+	} else {
+		if s.args.MasterCommand == "" || s.args.WorkerCommand == "" {
+			return fmt.Errorf("--command or --master-command/--worker-command must be set")
+		}
 	}
 	if s.args.MasterGPUCount < 0 || s.args.WorkerGPUCount < 0 {
 		return fmt.Errorf("--master-gpus/--worker-gpus is invalid")