We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 796f1c1 commit 3c8809bCopy full SHA for 3c8809b
1 file changed
sagemaker-core/src/sagemaker/core/model_registry.py
@@ -100,12 +100,9 @@ def get_model_package_args(
100
if model_life_cycle is not None:
101
model_package_args["model_life_cycle"] = model_life_cycle._to_request_dict()
102
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"]
+ original_req = {}
+ original_req["ModelCardContent"] = model_card.content
+ original_req["ModelCardStatus"] = model_card.model_card_status
109
model_package_args["model_card"] = original_req
110
return model_package_args
111
0 commit comments