Update error message

benieric · benieric · commit 038658377c17 · 2025-02-14T17:18:37.000-08:00
diff --git a/src/sagemaker/modules/train/model_trainer.py b/src/sagemaker/modules/train/model_trainer.py
@@ -835,8 +835,10 @@ def _prepare_train_script(
                 )
             execute_driver = EXECUTE_BASIC_SCRIPT_DRIVER
         else:
+            # This should never be reached, as the source_code should have been validated.
             raise ValueError(
-                f"Invalid configuration, please provide a valid SourceCode: {source_code}"
+                f"Unsupported SourceCode or DistributedConfig: {source_code}, {distributed}."
+                + f"Please provide a valid configuration with atleast one of 'command' or entry_script'."
             )
 
         train_script = TRAIN_SCRIPT_TEMPLATE.format(

Original file line number	Diff line number	Diff line change
`@@ -835,8 +835,10 @@ def _prepare_train_script(`
`835`	`835`	`)`
`836`	`836`	`execute_driver = EXECUTE_BASIC_SCRIPT_DRIVER`
`837`	`837`	`else:`
	`838`	`+ # This should never be reached, as the source_code should have been validated.`
`838`	`839`	`raise ValueError(`
`839`		`- f"Invalid configuration, please provide a valid SourceCode: {source_code}"`
	`840`	`+ f"Unsupported SourceCode or DistributedConfig: {source_code}, {distributed}."`
	`841`	`+ + f"Please provide a valid configuration with atleast one of 'command' or entry_script'."`
`840`	`842`	`)`
`841`	`843`
`842`	`844`	`train_script = TRAIN_SCRIPT_TEMPLATE.format(`