File tree 3 files changed +10
-4
lines changed
3 files changed +10
-4
lines changed Original file line number Diff line number Diff line change @@ -312,7 +312,7 @@ def register(
312
312
domain = None ,
313
313
):
314
314
"""Creates a model package for creating SageMaker models or listing on Marketplace.
315
-
315
+
316
316
Args:
317
317
content_types (list): The supported MIME types for the input data.
318
318
response_types (list): The supported MIME types for the output data.
Original file line number Diff line number Diff line change @@ -313,7 +313,9 @@ def register(
313
313
if model .model_data is None :
314
314
raise ValueError ("SageMaker Model Package cannot be created without model data." )
315
315
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
+ )
317
319
else :
318
320
container_def = [
319
321
{"Image" : image_uri or model .image_uri , "ModelDataUrl" : model .model_data }
Original file line number Diff line number Diff line change @@ -220,9 +220,13 @@ def __init__(
220
220
kwargs .pop ("output_kms_key" , None )
221
221
222
222
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
+ )
224
226
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
+ )]
226
230
227
231
register_model_step = _RegisterModelStep (
228
232
name = name ,
You can’t perform that action at this time.
0 commit comments