add s3 uri check to modeltrainer data source

mollyheamazon · mollyheamazon · commit 9ca799c17f08 · 2025-04-22T15:05:14.000-07:00
diff --git a/src/sagemaker/modules/train/model_trainer.py b/src/sagemaker/modules/train/model_trainer.py
@@ -407,7 +407,7 @@ def _validate_source_code(self, source_code: Optional[SourceCode]):
                         "If 'requirements' or 'entry_script' is provided in 'source_code', "
                         + "'source_dir' must also be provided.",
                     )
-                if not _is_valid_path(source_dir, path_type="Directory"):
+                if not _is_valid_path(source_dir, path_type="Directory") or _is_valid_s3_uri(source_dir, path_type="Directory"):
                     raise ValueError(
                         f"Invalid 'source_dir' path: {source_dir}. " + "Must be a valid directory.",
                     )

Original file line number	Diff line number	Diff line change
`@@ -407,7 +407,7 @@ def _validate_source_code(self, source_code: Optional[SourceCode]):`
`407`	`407`	`"If 'requirements' or 'entry_script' is provided in 'source_code', "`
`408`	`408`	`+ "'source_dir' must also be provided.",`
`409`	`409`	`)`
`410`		`- if not _is_valid_path(source_dir, path_type="Directory"):`
	`410`	`+ if not _is_valid_path(source_dir, path_type="Directory") or _is_valid_s3_uri(source_dir, path_type="Directory"):`
`411`	`411`	`raise ValueError(`
`412`	`412`	`f"Invalid 'source_dir' path: {source_dir}. " + "Must be a valid directory.",`
`413`	`413`	`)`