File tree Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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 = {
You can’t perform that action at this time.
0 commit comments