Skip to content

Commit f6cf94c

Browse files
committed
Account for both ModelCard and ModelPackageModelCard objects
1 parent 3c8809b commit f6cf94c

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

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

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
MODEL_PACKAGE_INFERENCE_SPECIFICATION_CONTAINERS_PATH,
1313
MODEL_PACKAGE_VALIDATION_PROFILES_PATH,
1414
)
15+
from sagemaker.core.resources import ModelPackageModelCard
1516
from botocore.exceptions import ClientError
1617
import logging
1718

@@ -101,7 +102,10 @@ def get_model_package_args(
101102
model_package_args["model_life_cycle"] = model_life_cycle._to_request_dict()
102103
if model_card is not None:
103104
original_req = {}
104-
original_req["ModelCardContent"] = model_card.content
105+
if isinstance(model_card, ModelPackageModelCard):
106+
original_req["ModelCardContent"] = model_card.model_card_content
107+
else:
108+
original_req["ModelCardContent"] = model_card.content
105109
original_req["ModelCardStatus"] = model_card.model_card_status
106110
model_package_args["model_card"] = original_req
107111
return model_package_args

0 commit comments

Comments
 (0)