Skip to content

Commit ec553fd

Browse files
leandrolcamposclaytonparnell
authored andcommitted
Fix style issues
1 parent 7dfe79a commit ec553fd

File tree

2 files changed

+4
-5
lines changed

2 files changed

+4
-5
lines changed

src/sagemaker/spark/processing.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -533,8 +533,7 @@ def _stage_submit_deps(self, submit_deps, input_channel_name):
533533
)
534534
else:
535535
input_channel_s3_uri = (
536-
f"{s3_prefix_uri}/{self._current_job_name}/"
537-
f"input/{input_channel_name}"
536+
f"{s3_prefix_uri}/{self._current_job_name}/input/{input_channel_name}"
538537
)
539538
logger.info(
540539
"Uploading dependencies from tmpdir %s to S3 %s", tmpdir, input_channel_s3_uri

tests/unit/sagemaker/spark/test_processing.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -355,7 +355,7 @@ def test_stage_configuration(mock_s3_upload, mock_bytesIO, config, expected, sag
355355
"dependency_location": None,
356356
"submit_deps": ["s3"],
357357
"input_channel_name": None,
358-
},
358+
},
359359
ValueError,
360360
),
361361
(
@@ -412,7 +412,7 @@ def test_stage_configuration(mock_s3_upload, mock_bytesIO, config, expected, sag
412412
},
413413
(
414414
"s3://codebucket/someprefix/None/input/channelName",
415-
"/opt/ml/processing/input/channelName"
415+
"/opt/ml/processing/input/channelName",
416416
),
417417
),
418418
(
@@ -433,7 +433,7 @@ def test_stage_configuration(mock_s3_upload, mock_bytesIO, config, expected, sag
433433
},
434434
(
435435
"s3://codebucket/someprefix/None/input/channelName",
436-
"/opt/ml/processing/input/channelName"
436+
"/opt/ml/processing/input/channelName",
437437
),
438438
),
439439
],

0 commit comments

Comments
 (0)