diff --git a/src/sagemaker/processing.py b/src/sagemaker/processing.py index 163a06e2d1..4cf5c8c432 100644 --- a/src/sagemaker/processing.py +++ b/src/sagemaker/processing.py @@ -213,11 +213,9 @@ def _normalize_inputs(self, inputs=None): # and save the S3 uri in the ProcessingInput source. parse_result = urlparse(file_input.source) if parse_result.scheme != "s3": - desired_s3_uri = os.path.join( - "s3://", + desired_s3_uri = "s3://{}/{}/input/{}".format( self.sagemaker_session.default_bucket(), self._current_job_name, - "input", file_input.input_name, ) s3_uri = S3Uploader.upload( @@ -259,11 +257,9 @@ def _normalize_outputs(self, outputs=None): # If the output's destination is not an s3_uri, create one. parse_result = urlparse(output.destination) if parse_result.scheme != "s3": - s3_uri = os.path.join( - "s3://", + s3_uri = "s3://{}/{}/output/{}".format( self.sagemaker_session.default_bucket(), self._current_job_name, - "output", output.output_name, ) output.destination = s3_uri @@ -475,11 +471,9 @@ def _upload_code(self, code): str: The S3 URI of the uploaded file or directory. """ - desired_s3_uri = os.path.join( - "s3://", + desired_s3_uri = "s3://{}/{}/input/{}".format( self.sagemaker_session.default_bucket(), self._current_job_name, - "input", self._CODE_CONTAINER_INPUT_NAME, ) return S3Uploader.upload(