fix: fix kmeans test deletion sequence, increment lineage statics

Author: mufaddal-rohawala

src/sagemaker/session.py

@@ -3565,6 +3565,7 @@ def endpoint_from_production_variants(
         if data_capture_config_dict is not None:
             config_options["DataCaptureConfig"] = data_capture_config_dict
 
+        LOGGER.info(f"Creating Endpoint Config with name: {name}")
         self.sagemaker_client.create_endpoint_config(**config_options)
 
         return self.create_endpoint(endpoint_name=name, config_name=name, tags=tags, wait=wait)

tests/integ/sagemaker/lineage/conftest.py

@@ -36,8 +36,8 @@
 from tests.integ.sagemaker.lineage.helpers import name, names
 
 SLEEP_TIME_SECONDS = 1
-STATIC_PIPELINE_NAME = "SdkIntegTestStaticPipeline15"
-STATIC_ENDPOINT_NAME = "SdkIntegTestStaticEndpoint15"
+STATIC_PIPELINE_NAME = "SdkIntegTestStaticPipeline16"
+STATIC_ENDPOINT_NAME = "SdkIntegTestStaticEndpoint16"
 
 
 @pytest.fixture

tests/integ/test_kmeans.py

@@ -76,11 +76,10 @@ def test_kmeans(sagemaker_session, cpu_instance_type, training_set):
         for record in result:
             assert record.label["closest_cluster"] is not None
             assert record.label["distance_to_cluster"] is not None
-
-    predictor.delete_model()
-    with pytest.raises(Exception) as exception:
-        sagemaker_session.sagemaker_client.describe_model(ModelName=model.name)
-        assert "Could not find model" in str(exception.value)
+        predictor.delete_model()
+        with pytest.raises(Exception) as exception:
+            sagemaker_session.sagemaker_client.describe_model(ModelName=model.name)
+            assert "Could not find model" in str(exception.value)
 
 
 def test_async_kmeans(sagemaker_session, cpu_instance_type, training_set):