Skip to content

Commit fad0fff

Browse files
committed
Fix unit test
1 parent b764513 commit fad0fff

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

src/sagemaker/session.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7685,7 +7685,7 @@ def get_create_model_package_request(
76857685
if model_card is not None:
76867686
request_dict["ModelCard"] = model_card
76877687
if model_life_cycle is not None:
7688-
request_dict["ModelLifeCycle"] = model_life_cycle._to_request_dict()
7688+
request_dict["ModelLifeCycle"] = model_life_cycle
76897689
return request_dict
76907690

76917691

tests/unit/test_estimator.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4369,7 +4369,6 @@ def test_register_default_image(sagemaker_session):
43694369
stage_status="In-Progress",
43704370
stage_description="Sending for Staging Verification",
43714371
)
4372-
update_model_life_cycle_req = update_model_life_cycle._to_request_dict()
43734372

43744373
estimator.register(
43754374
content_types=content_types,
@@ -4384,7 +4383,7 @@ def test_register_default_image(sagemaker_session):
43844383
nearest_model_name=nearest_model_name,
43854384
data_input_configuration=data_input_config,
43864385
model_card=model_card,
4387-
model_life_cycle=update_model_life_cycle_req,
4386+
model_life_cycle=update_model_life_cycle,
43884387
)
43894388
sagemaker_session.create_model.assert_not_called()
43904389
exp_model_card = {

0 commit comments

Comments
 (0)