File tree Expand file tree Collapse file tree
sagemaker-core/src/sagemaker/core Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1212 MODEL_PACKAGE_INFERENCE_SPECIFICATION_CONTAINERS_PATH ,
1313 MODEL_PACKAGE_VALIDATION_PROFILES_PATH ,
1414)
15+ from sagemaker .core .resources import ModelPackageModelCard
1516from botocore .exceptions import ClientError
1617import 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
You can’t perform that action at this time.
0 commit comments