fix: update_model_card input types

selvask-aws · selvask-aws · commit 9366c788cbdb · 2024-06-26T00:45:23.000+05:30
diff --git a/src/sagemaker/model.py b/src/sagemaker/model.py
@@ -2428,7 +2428,7 @@ def add_inference_specification(
 
         sagemaker_session.sagemaker_client.update_model_package(**model_package_update_args)
 
-    def update_model_card(self, model_card: ModelCard | ModelPackageModelCard):
+    def update_model_card(self, model_card: Union[ModelCard | ModelPackageModelCard]):
         """Updates Created model card content which created with model package
 
         Args:
