Fix broken tags

Author: martinRenou

tests/unit/sagemaker/model/test_model.py

@@ -361,7 +361,7 @@ def test_create_sagemaker_model_tags(prepare_container_def, sagemaker_session):
 
     model = Model(MODEL_IMAGE, MODEL_DATA, name=MODEL_NAME, sagemaker_session=sagemaker_session)
 
-    tags = {"Key": "foo", "Value": "bar"}
+    tags = [{"Key": "foo", "Value": "bar"}]
     model._create_sagemaker_model(INSTANCE_TYPE, tags=tags)
 
     sagemaker_session.create_model.assert_called_with(

tests/unit/sagemaker/model/test_model_package.py

@@ -201,7 +201,7 @@ def test_create_sagemaker_model_include_tags(sagemaker_session):
     env_key = "env_key"
     env_value = "env_value"
     environment = {env_key: env_value}
-    tags = {"Key": "foo", "Value": "bar"}
+    tags = [{"Key": "foo", "Value": "bar"}]
 
     model_package = ModelPackage(
         role="role",
@@ -314,7 +314,7 @@ def test_model_package_create_transformer_with_product_id(sagemaker_session):
 
 @patch("sagemaker.model.ModelPackage.update_approval_status")
 def test_model_package_auto_approve_on_deploy(update_approval_status, sagemaker_session):
-    tags = {"Key": "foo", "Value": "bar"}
+    tags = [{"Key": "foo", "Value": "bar"}]
     model_package = ModelPackage(
         role="role",
         model_package_arn=MODEL_PACKAGE_VERSIONED_ARN,

tests/unit/sagemaker/tensorflow/test_estimator.py

@@ -324,7 +324,7 @@ def test_transformer_creation_with_optional_args(
     env = {"foo": "bar"}
     max_concurrent_transforms = 3
     max_payload = 100
-    tags = {"Key": "foo", "Value": "bar"}
+    tags = [{"Key": "foo", "Value": "bar"}]
     new_role = "role"
     vpc_config = {"Subnets": ["1234"], "SecurityGroupIds": ["5678"]}
     model_name = "model-name"

tests/unit/test_predictor.py

@@ -307,7 +307,7 @@ def test_update_endpoint_all_args(name_from_base, production_variant):
     new_instance_type = "ml.c4.xlarge"
     new_accelerator_type = "ml.eia1.medium"
     new_model_name = "new-model"
-    new_tags = {"Key": "foo", "Value": "bar"}
+    new_tags = [{"Key": "foo", "Value": "bar"}]
     new_kms_key = "new-key"
     new_data_capture_config_dict = {}
 

tests/unit/test_predictor_async.py

@@ -404,7 +404,7 @@ def test_update_endpoint_all_args():
     new_instance_type = "ml.c4.xlarge"
     new_accelerator_type = "ml.eia1.medium"
     new_model_name = "new-model"
-    new_tags = {"Key": "foo", "Value": "bar"}
+    new_tags = [{"Key": "foo", "Value": "bar"}]
     new_kms_key = "new-key"
     new_data_capture_config_dict = {}
 

tests/unit/test_transformer.py

@@ -249,7 +249,7 @@ def test_transformer_init_optional_params(sagemaker_session):
     accept = "text/csv"
     max_concurrent_transforms = 100
     max_payload = 100
-    tags = {"Key": "foo", "Value": "bar"}
+    tags = [{"Key": "foo", "Value": "bar"}]
     env = {"FOO": "BAR"}
 
     transformer = Transformer(