fix: unit test cases

Author: selvask-aws

src/sagemaker/model.py

@@ -549,11 +549,7 @@ def register(
             model_package_group_name = utils.base_name_from_image(
                 self.image_uri, default_base_name=ModelPackage.__name__
             )
-        if (
-            model_package_group_name is not None
-            and self.model_type is not None
-            and self.model_type is not JumpStartModelType.PROPRIETARY
-        ):
+        if model_package_group_name is not None and model_type is not JumpStartModelType.PROPRIETARY:
             container_def = self.prepare_container_def(accept_eula=accept_eula)
             container_def = update_container_with_inference_params(
                 framework=framework,

tests/unit/test_estimator.py

@@ -4402,7 +4402,10 @@ def test_register_default_image_without_instance_type_args(sagemaker_session):
     framework = "TENSORFLOW"
     framework_version = "2.9"
     nearest_model_name = "resnet50"
-
+    model_card = {
+        'ModelCardStatus': ModelCardStatusEnum.DRAFT,
+        'ModelCardContent': '{}'
+    }
     estimator.register(
         content_types=content_types,
         response_types=response_types,
@@ -4425,6 +4428,7 @@ def test_register_default_image_without_instance_type_args(sagemaker_session):
         "marketplace_cert": False,
         "sample_payload_url": sample_payload_url,
         "task": task,
+        "model_card": model_card,
     }
     sagemaker_session.create_model_package_from_containers.assert_called_with(
         **expected_create_model_package_request
@@ -4454,6 +4458,10 @@ def test_register_inference_image(sagemaker_session):
     framework = "TENSORFLOW"
     framework_version = "2.9"
     nearest_model_name = "resnet50"
+    model_card = {
+        'ModelCardStatus': ModelCardStatusEnum.DRAFT,
+        'ModelCardContent': '{}'
+    }
 
     estimator.register(
         content_types=content_types,
@@ -4480,6 +4488,7 @@ def test_register_inference_image(sagemaker_session):
         "marketplace_cert": False,
         "sample_payload_url": sample_payload_url,
         "task": task,
+        "model_card": model_card,
     }
     sagemaker_session.create_model_package_from_containers.assert_called_with(
         **expected_create_model_package_request