File tree 3 files changed +10
-5
lines changed
3 files changed +10
-5
lines changed Original file line number Diff line number Diff line change @@ -4516,12 +4516,15 @@ def get_create_model_package_request(
4516
4516
else :
4517
4517
if not all ([inference_instances , transform_instances ]):
4518
4518
raise ValueError (
4519
- "inference_instances and transform_instances " "must be provided if model_package_group_name is not present."
4519
+ "inference_instances and transform_instances "
4520
+ "must be provided if model_package_group_name is not present."
4520
4521
)
4521
- inference_specification .update ({
4522
- "SupportedRealtimeInferenceInstanceTypes" : inference_instances ,
4523
- "SupportedTransformInstanceTypes" : transform_instances ,
4524
- })
4522
+ inference_specification .update (
4523
+ {
4524
+ "SupportedRealtimeInferenceInstanceTypes" : inference_instances ,
4525
+ "SupportedTransformInstanceTypes" : transform_instances ,
4526
+ }
4527
+ )
4525
4528
request_dict ["InferenceSpecification" ] = inference_specification
4526
4529
request_dict ["CertifyForMarketplace" ] = marketplace_cert
4527
4530
request_dict ["ModelApprovalStatus" ] = approval_status
Original file line number Diff line number Diff line change @@ -299,6 +299,7 @@ def test_pipeline_session_context_for_model_step_with_one_instance_types(
299
299
300
300
assert register_step_args .create_model_package_request == expected_output
301
301
302
+
302
303
def test_pipeline_session_context_for_model_step_without_model_package_group_name (
303
304
pipeline_session_mock ,
304
305
):
Original file line number Diff line number Diff line change @@ -2359,6 +2359,7 @@ def test_create_model_package_from_containers_incomplete_args(sagemaker_session)
2359
2359
"must be provided if containers is present." == str (error )
2360
2360
)
2361
2361
2362
+
2362
2363
def test_create_model_package_from_containers_without_model_package_group_name (sagemaker_session ):
2363
2364
model_package_name = "sagemaker-model-package"
2364
2365
containers = ["dummy-container" ]
You can’t perform that action at this time.
0 commit comments