Skip to content

Commit 28a2b9f

Browse files
committed
change: format files
1 parent 2072932 commit 28a2b9f

File tree

4 files changed

+4
-12
lines changed

4 files changed

+4
-12
lines changed

src/sagemaker/estimator.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1407,7 +1407,5 @@ def _s3_uri_without_prefix_from_input(input_data):
14071407
if isinstance(input_data, s3_input):
14081408
return _s3_uri_prefix("training", input_data)
14091409
raise ValueError(
1410-
"Unrecognized type for S3 input data config - not str or s3_input: {}".format(
1411-
input_data
1412-
)
1410+
"Unrecognized type for S3 input data config - not str or s3_input: {}".format(input_data)
14131411
)

src/sagemaker/fw_utils.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -180,9 +180,7 @@ def create_image_uri(
180180
return "{}/{}:{}".format(
181181
get_ecr_image_uri_prefix(account, region), MERGED_FRAMEWORKS_REPO_MAP[framework], tag
182182
)
183-
return "{}/sagemaker-{}:{}".format(
184-
get_ecr_image_uri_prefix(account, region), framework, tag
185-
)
183+
return "{}/sagemaker-{}:{}".format(get_ecr_image_uri_prefix(account, region), framework, tag)
186184

187185

188186
def _accelerator_type_valid_for_framework(

src/sagemaker/job.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -196,9 +196,7 @@ def _format_model_uri_input(model_uri, validate_uri=True):
196196
distribution="FullyReplicated",
197197
content_type="application/x-sagemaker-model",
198198
)
199-
if (
200-
isinstance(model_uri, string_types) and validate_uri and model_uri.startswith("file://")
201-
):
199+
if isinstance(model_uri, string_types) and validate_uri and model_uri.startswith("file://"):
202200
return file_input(model_uri)
203201
if isinstance(model_uri, string_types) and validate_uri:
204202
raise ValueError(

src/sagemaker/local/data.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -83,9 +83,7 @@ def get_batch_strategy_instance(strategy, splitter):
8383
return SingleRecordStrategy(splitter)
8484
if strategy == "MultiRecord":
8585
return MultiRecordStrategy(splitter)
86-
raise ValueError(
87-
'Invalid Batch Strategy: %s - Valid Strategies: "SingleRecord", "MultiRecord"'
88-
)
86+
raise ValueError('Invalid Batch Strategy: %s - Valid Strategies: "SingleRecord", "MultiRecord"')
8987

9088

9189
class DataSource(with_metaclass(ABCMeta, object)):

0 commit comments

Comments
 (0)