fix: adding Unit tests for resourcekey and tags for api in config for intelligent defaults

Added Unit tests to test the config injection
for SAGEMAKER_CONFIG_ENDPOINT

Fixed a couple of unit tests

Author: akrishna1995

src/sagemaker/session.py

@@ -88,6 +88,7 @@
     ENDPOINT_CONFIG,
     ENDPOINT_CONFIG_DATA_CAPTURE_PATH,
     ENDPOINT_CONFIG_ASYNC_INFERENCE_PATH,
+    ENDPOINT,
     ENDPOINT_TAGS_PATH,
     SAGEMAKER,
     FEATURE_GROUP,
@@ -3968,7 +3969,7 @@ def create_endpoint(self, endpoint_name, config_name, tags=None, wait=True):
         tags = tags or []
         tags = _append_project_tags(tags)
         tags = self._append_sagemaker_config_tags(
-            tags, "{}.{}.{}".format(SAGEMAKER, ENDPOINT_TAGS_PATH, TAGS)
+            tags, "{}.{}.{}".format(SAGEMAKER, ENDPOINT, TAGS)
         )
 
         self.sagemaker_client.create_endpoint(
@@ -4468,9 +4469,6 @@ def endpoint_from_model_data(
         model_vpc_config = vpc_utils.sanitize(model_vpc_config)
         endpoint_config_tags = _append_project_tags(tags)
         endpoint_tags = _append_project_tags(tags)
-        endpoint_tags = self._append_sagemaker_config_tags(
-            endpoint_tags, "{}.{}.{}".format(SAGEMAKER, ENDPOINT_TAGS_PATH, TAGS)
-        )
         endpoint_config_tags = self._append_sagemaker_config_tags(
             endpoint_config_tags, "{}.{}.{}".format(SAGEMAKER, ENDPOINT_CONFIG, TAGS)
         )
@@ -4571,9 +4569,6 @@ def endpoint_from_production_variants(
         endpoint_config_tags = _append_project_tags(tags)
         endpoint_tags = _append_project_tags(tags)
 
-        endpoint_tags = self._append_sagemaker_config_tags(
-            endpoint_tags, "{}.{}.{}".format(SAGEMAKER, ENDPOINT_TAGS_PATH, TAGS)
-        )
         endpoint_config_tags = self._append_sagemaker_config_tags(
             endpoint_config_tags, "{}.{}.{}".format(SAGEMAKER, ENDPOINT_CONFIG, TAGS)
         )

tests/unit/test_session.py

@@ -2552,7 +2552,7 @@ def test_create_edge_packaging_with_sagemaker_config_injection(sagemaker_session
         "OutputConfig"
     ]["KmsKeyId"]
     expected_tags = SAGEMAKER_CONFIG_EDGE_PACKAGING_JOB["SageMaker"]["EdgePackagingJob"]["Tags"]
-    expected_resource_key = SAGEMAKER_CONFIG_EDGE_PACKAGING_JOB["SageMaker"]["EdgePackagingJob"]["ResourceKey"]
+    expected_resource_key = (SAGEMAKER_CONFIG_EDGE_PACKAGING_JOB["SageMaker"]["EdgePackagingJob"]["ResourceKey"],)
     sagemaker_session.sagemaker_client.create_edge_packaging_job.assert_called_with(
         RoleArn=expected_role_arn,  # provided from config
         OutputConfig={
@@ -3058,7 +3058,7 @@ def test_endpoint_from_production_variants_with_sagemaker_config_injection(
     sagemaker_session.sagemaker_client.create_endpoint.assert_called_with(
         EndpointConfigName="some-endpoint",
         EndpointName="some-endpoint",
-        Tags=expected_tags,  # from config
+        Tags=[],
     )
 
 
@@ -3126,7 +3126,7 @@ def test_endpoint_from_production_variants_with_sagemaker_config_injection_parti
     sagemaker_session.sagemaker_client.create_endpoint.assert_called_with(
         EndpointConfigName="some-endpoint",
         EndpointName="some-endpoint",
-        Tags=expected_tags,  # from config
+        Tags=[],
     )
 
 
@@ -3187,7 +3187,7 @@ def test_endpoint_from_production_variants_with_sagemaker_config_injection_no_km
     sagemaker_session.sagemaker_client.create_endpoint.assert_called_with(
         EndpointConfigName="some-endpoint",
         EndpointName="some-endpoint",
-        Tags=expected_tags,  # from config
+        Tags=[],
     )
 
 
@@ -3268,7 +3268,7 @@ def test_endpoint_from_production_variants_with_sagemaker_config_injection_tags(
         EndpointConfigName="some-endpoint", EndpointName="some-endpoint", Tags=expected_tags
     )
     sagemaker_session.sagemaker_client.create_endpoint_config.assert_called_with(
-        EndpointConfigName="some-endpoint", ProductionVariants=pvs, Tags=[]
+        EndpointConfigName="some-endpoint", ProductionVariants=pvs
     )
 
 
@@ -3319,11 +3319,12 @@ def test_endpoint_from_production_variants_with_accelerator_type_sagemaker_confi
     )
     ims.sagemaker_client.describe_endpoint_config = Mock(side_effect=ex)
     expected_tags = SAGEMAKER_CONFIG_ENDPOINT["SageMaker"]["Endpoint"]["Tags"]
+    sagemaker_session.endpoint_from_production_variants("some-endpoint", pvs)
     sagemaker_session.sagemaker_client.create_endpoint.assert_called_with(
         EndpointConfigName="some-endpoint", EndpointName="some-endpoint", Tags=expected_tags
     )
     sagemaker_session.sagemaker_client.create_endpoint_config.assert_called_with(
-        EndpointConfigName="some-endpoint", ProductionVariants=pvs, Tags=[]
+        EndpointConfigName="some-endpoint", ProductionVariants=pvs
     )
 
 
@@ -3395,7 +3396,7 @@ def test_endpoint_from_production_variants_with_serverless_inference_config_sage
         EndpointConfigName="some-endpoint", EndpointName="some-endpoint", Tags=expected_tags
     )
     sagemaker_session.sagemaker_client.create_endpoint_config.assert_called_with(
-        EndpointConfigName="some-endpoint", ProductionVariants=pvs, Tags=[]
+        EndpointConfigName="some-endpoint", ProductionVariants=pvs
     )