Skip to content

Commit 5ffe718

Browse files
Fix bug forcing uploaded tar to be named sourcedir
1 parent 0914f17 commit 5ffe718

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

src/sagemaker/processing.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -1679,10 +1679,8 @@ def _pack_and_upload_code(
16791679
)
16801680

16811681
# Upload the bootstrapping code as s3://.../jobname/source/runproc.sh.
1682-
entrypoint_s3_uri = estimator.uploaded_code.s3_prefix.replace(
1683-
"sourcedir.tar.gz",
1684-
"runproc.sh",
1685-
)
1682+
entrypoint_s3_uri = estimator.uploaded_code.s3_prefix.rsplit('/', 1)[0] + '/runproc.sh'
1683+
16861684
script = estimator.uploaded_code.script_name
16871685
s3_runproc_sh = S3Uploader.upload_string_as_file_body(
16881686
self._generate_framework_script(script),

0 commit comments

Comments
 (0)