Skip to content

Commit 95af758

Browse files
committed
fix: black-check and flake8 errors
1 parent bcb3425 commit 95af758

File tree

3 files changed

+10
-5
lines changed

3 files changed

+10
-5
lines changed

src/sagemaker/session.py

+8-5
Original file line numberDiff line numberDiff line change
@@ -4516,12 +4516,15 @@ def get_create_model_package_request(
45164516
else:
45174517
if not all([inference_instances, transform_instances]):
45184518
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."
45204521
)
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+
)
45254528
request_dict["InferenceSpecification"] = inference_specification
45264529
request_dict["CertifyForMarketplace"] = marketplace_cert
45274530
request_dict["ModelApprovalStatus"] = approval_status

tests/unit/sagemaker/workflow/test_pipeline_session.py

+1
Original file line numberDiff line numberDiff line change
@@ -299,6 +299,7 @@ def test_pipeline_session_context_for_model_step_with_one_instance_types(
299299

300300
assert register_step_args.create_model_package_request == expected_output
301301

302+
302303
def test_pipeline_session_context_for_model_step_without_model_package_group_name(
303304
pipeline_session_mock,
304305
):

tests/unit/test_session.py

+1
Original file line numberDiff line numberDiff line change
@@ -2359,6 +2359,7 @@ def test_create_model_package_from_containers_incomplete_args(sagemaker_session)
23592359
"must be provided if containers is present." == str(error)
23602360
)
23612361

2362+
23622363
def test_create_model_package_from_containers_without_model_package_group_name(sagemaker_session):
23632364
model_package_name = "sagemaker-model-package"
23642365
containers = ["dummy-container"]

0 commit comments

Comments
 (0)