Skip to content

Commit 3a8e471

Browse files
GroovyDanclaytonparnell
authored andcommitted
fix: do not use in airflow prepare_framework function
1 parent f2d5e41 commit 3a8e471

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/workflow/airflow.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,12 +37,12 @@ def prepare_framework(estimator, s3_operations):
3737
"""
3838
if estimator.code_location is not None:
3939
bucket, key = s3.parse_s3_url(estimator.code_location)
40-
key = os.path.join(key, estimator._current_job_name, "source", "sourcedir.tar.gz")
40+
key = "/".join([key, estimator._current_job_name, "source", "sourcedir.tar.gz"])
4141
elif estimator.uploaded_code is not None:
4242
bucket, key = s3.parse_s3_url(estimator.uploaded_code.s3_prefix)
4343
else:
4444
bucket = estimator.sagemaker_session._default_bucket
45-
key = os.path.join(estimator._current_job_name, "source", "sourcedir.tar.gz")
45+
key = "/".join([estimator._current_job_name, "source", "sourcedir.tar.gz"])
4646

4747
script = os.path.basename(estimator.entry_point)
4848

0 commit comments

Comments
 (0)