Skip to content

Commit 3c8809b

Browse files
committed
Fix model registration with a model card
1 parent 796f1c1 commit 3c8809b

1 file changed

Lines changed: 3 additions & 6 deletions

File tree

sagemaker-core/src/sagemaker/core/model_registry.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -100,12 +100,9 @@ def get_model_package_args(
100100
if model_life_cycle is not None:
101101
model_package_args["model_life_cycle"] = model_life_cycle._to_request_dict()
102102
if model_card is not None:
103-
original_req = model_card._create_request_args()
104-
if original_req.get("ModelCardName") is not None:
105-
del original_req["ModelCardName"]
106-
if original_req.get("Content") is not None:
107-
original_req["ModelCardContent"] = original_req["Content"]
108-
del original_req["Content"]
103+
original_req = {}
104+
original_req["ModelCardContent"] = model_card.content
105+
original_req["ModelCardStatus"] = model_card.model_card_status
109106
model_package_args["model_card"] = original_req
110107
return model_package_args
111108

0 commit comments

Comments
 (0)