Skip to content

Commit d5a913a

Browse files
Fix style issues
1 parent 549a719 commit d5a913a

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
@@ -523,8 +523,7 @@ def _stage_submit_deps(self, submit_deps, input_channel_name):
523523
)
524524
else:
525525
input_channel_s3_uri = (
526-
f"{s3_prefix_uri}/{self._current_job_name}/"
527-
f"input/{input_channel_name}"
526+
f"{s3_prefix_uri}/{self._current_job_name}/input/{input_channel_name}"
528527
)
529528
logger.info(
530529
"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
@@ -353,7 +353,7 @@ def test_stage_configuration(mock_s3_upload, mock_bytesIO, config, expected, sag
353353
"dependency_location": None,
354354
"submit_deps": ["s3"],
355355
"input_channel_name": None,
356-
},
356+
},
357357
ValueError,
358358
),
359359
(
@@ -410,7 +410,7 @@ def test_stage_configuration(mock_s3_upload, mock_bytesIO, config, expected, sag
410410
},
411411
(
412412
"s3://codebucket/someprefix/None/input/channelName",
413-
"/opt/ml/processing/input/channelName"
413+
"/opt/ml/processing/input/channelName",
414414
),
415415
),
416416
(
@@ -431,7 +431,7 @@ def test_stage_configuration(mock_s3_upload, mock_bytesIO, config, expected, sag
431431
},
432432
(
433433
"s3://codebucket/someprefix/None/input/channelName",
434-
"/opt/ml/processing/input/channelName"
434+
"/opt/ml/processing/input/channelName",
435435
),
436436
),
437437
],

0 commit comments

Comments
 (0)