File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -7509,7 +7509,7 @@ def get_model_package_args(
7509
7509
if source_uri is not None :
7510
7510
model_package_args ["source_uri" ] = source_uri
7511
7511
if model_life_cycle is not None :
7512
- model_package_args ["model_life_cycle" ] = model_life_cycle
7512
+ model_package_args ["model_life_cycle" ] = model_life_cycle . _to_request_dict ()
7513
7513
if model_card is not None :
7514
7514
original_req = model_card ._create_request_args ()
7515
7515
if original_req .get ("ModelCardName" ) is not None :
@@ -7685,7 +7685,7 @@ def get_create_model_package_request(
7685
7685
if model_card is not None :
7686
7686
request_dict ["ModelCard" ] = model_card
7687
7687
if model_life_cycle is not None :
7688
- request_dict ["ModelLifeCycle" ] = model_life_cycle
7688
+ request_dict ["ModelLifeCycle" ] = model_life_cycle . _to_request_dict ()
7689
7689
return request_dict
7690
7690
7691
7691
You can’t perform that action at this time.
0 commit comments