Skip to content

Commit effd3ee

Browse files
committed
fix: fix pylint errors
1 parent b57ad62 commit effd3ee

File tree

3 files changed

+10
-4
lines changed

3 files changed

+10
-4
lines changed

src/sagemaker/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -312,7 +312,7 @@ def register(
312312
domain=None,
313313
):
314314
"""Creates a model package for creating SageMaker models or listing on Marketplace.
315-
315+
316316
Args:
317317
content_types (list): The supported MIME types for the input data.
318318
response_types (list): The supported MIME types for the output data.

src/sagemaker/pipeline.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,9 @@ def register(
313313
if model.model_data is None:
314314
raise ValueError("SageMaker Model Package cannot be created without model data.")
315315
if model_package_group_name is not None:
316-
container_def = self.pipeline_container_def(inference_instances[0] if inference_instances else None)
316+
container_def = self.pipeline_container_def(
317+
inference_instances[0] if inference_instances else None
318+
)
317319
else:
318320
container_def = [
319321
{"Image": image_uri or model.image_uri, "ModelDataUrl": model.model_data}

src/sagemaker/workflow/step_collections.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -220,9 +220,13 @@ def __init__(
220220
kwargs.pop("output_kms_key", None)
221221

222222
if isinstance(model, PipelineModel):
223-
self.container_def_list = model.pipeline_container_def(inference_instances[0] if inference_instances else None)
223+
self.container_def_list = model.pipeline_container_def(
224+
inference_instances[0] if inference_instances else None
225+
)
224226
elif isinstance(model, Model):
225-
self.container_def_list = [model.prepare_container_def(inference_instances[0] if inference_instances else None)]
227+
self.container_def_list = [model.prepare_container_def(
228+
inference_instances[0] if inference_instances else None
229+
)]
226230

227231
register_model_step = _RegisterModelStep(
228232
name=name,

0 commit comments

Comments
 (0)