From 81baa4b5fc9e3d06144e6821d139405cfd73cdea Mon Sep 17 00:00:00 2001 From: Ajay Karpur Date: Thu, 12 Dec 2019 12:47:02 -0800 Subject: [PATCH] Fix bug with specifying S3 uri as code location --- src/sagemaker/processing.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/sagemaker/processing.py b/src/sagemaker/processing.py index 07c4fb53a3..e8b723a711 100644 --- a/src/sagemaker/processing.py +++ b/src/sagemaker/processing.py @@ -370,7 +370,10 @@ def run( self._current_job_name = self._generate_current_job_name(job_name=job_name) user_script_name = self._get_user_script_name(code) - user_code_s3_uri = self._upload_code(code) + if urlparse(code).scheme == "s3": + user_code_s3_uri = code + else: + user_code_s3_uri = self._upload_code(code) inputs_with_code = self._convert_code_and_add_to_inputs(inputs, user_code_s3_uri) self._set_entrypoint(self.command, user_script_name)