Skip to content

Commit 0a7c936

Browse files
author
Joseph Zhang
committed
Fix formatting.
1 parent ba7b9af commit 0a7c936

File tree

1 file changed

+8
-14
lines changed

1 file changed

+8
-14
lines changed

tests/integ/test_inference_recommender.py

+8-14
Original file line numberDiff line numberDiff line change
@@ -461,9 +461,7 @@ def test_deploy_inference_recommendation_id_with_registered_model_sklearn(
461461
JobName=ir_job_name
462462
)
463463

464-
rec_id = get_realtime_recommendation_id(
465-
recommendation_list=rec_res["InferenceRecommendations"]
466-
)
464+
rec_id = get_realtime_recommendation_id(recommendation_list=rec_res["InferenceRecommendations"])
467465

468466
with timeout(minutes=45):
469467
try:
@@ -540,15 +538,11 @@ def poll_for_deployment_recommendation(created_base_model, sagemaker_session):
540538
except Exception as e:
541539
created_base_model.delete_model()
542540
raise e
543-
544-
def get_realtime_recommendation_id(recommendation_list):
545-
"""Search recommendation based on recommendation id"""
546-
next(
547-
(
548-
rec["RecommendationId"]
549-
for rec in recommendation_list
550-
if "InstanceType" in rec
551-
),
552-
None,
553-
)
554541

542+
543+
def get_realtime_recommendation_id(recommendation_list):
544+
"""Search recommendation based on recommendation id"""
545+
next(
546+
(rec["RecommendationId"] for rec in recommendation_list if "InstanceType" in rec),
547+
None,
548+
)

0 commit comments

Comments
 (0)