Fix tests

Author: martinRenou

src/sagemaker/workflow/notebook_job_step.py

@@ -358,7 +358,7 @@ def _prepare_tags(self):
         This function converts the custom tags into training API required format and also
         attach the system tags.
         """
-        custom_tags = format_tags(self.tags)
+        custom_tags = format_tags(self.tags) or []
         system_tags = [
             {"Key": "sagemaker:name", "Value": self.notebook_job_name},
             {"Key": "sagemaker:notebook-name", "Value": os.path.basename(self.input_notebook)},

tests/unit/test_session.py

@@ -3596,7 +3596,7 @@ def test_create_endpoint_config_with_tags_list(sagemaker_session):
 
 
 def test_create_endpoint_config_with_tags_dict(sagemaker_session):
-    tags = {"TagtestKey", "TagtestValue"}
+    tags = {"TagtestKey": "TagtestValue"}
     call_tags = [{"Key": "TagtestKey", "Value": "TagtestValue"}]
 
     sagemaker_session.create_endpoint_config(

tests/unit/test_transformer.py

@@ -573,7 +573,7 @@ def test_start_new(prepare_data_processing, load_config, sagemaker_session):
     strategy = "MultiRecord"
     max_concurrent_transforms = 100
     max_payload = 100
-    tags = {"Key": "foo", "Value": "bar"}
+    tags = [{"Key": "foo", "Value": "bar"}]
     env = {"FOO": "BAR"}
 
     transformer = Transformer(