Skip to content

Commit b764513

Browse files
committed
Fix: Object of type ModelLifeCycle is not JSON serializable
1 parent 5198f28 commit b764513

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/session.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7509,7 +7509,7 @@ def get_model_package_args(
75097509
if source_uri is not None:
75107510
model_package_args["source_uri"] = source_uri
75117511
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()
75137513
if model_card is not None:
75147514
original_req = model_card._create_request_args()
75157515
if original_req.get("ModelCardName") is not None:
@@ -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
7688+
request_dict["ModelLifeCycle"] = model_life_cycle._to_request_dict()
76897689
return request_dict
76907690

76917691

0 commit comments

Comments
 (0)