Fix Alabtross Inference component tests

Author: Pravali Uppugunduri

src/sagemaker/serve/builder/model_builder.py

@@ -1983,7 +1983,6 @@ def deploy(
         """
         if not hasattr(self, "built_model") and not hasattr(self, "_deployables"):
             raise ValueError("Model needs to be built before deploying")
-        endpoint_name = unique_name_from_base("endpoint-name")
         if not update_endpoint:
             endpoint_name = unique_name_from_base(endpoint_name)
 

tests/integ/sagemaker/serve/test_serve_model_builder_inference_component_happy.py

@@ -40,6 +40,7 @@
 
 LLAMA_2_7B_JS_ID = "meta-textgeneration-llama-2-7b"
 LLAMA_IC_NAME = "llama2-mb-ic"
+INSTANCE_TYPE = "ml.g5.24xlarge"
 
 
 @pytest.fixture
@@ -65,6 +66,7 @@ def test_model_builder_ic_sagemaker_endpoint(
     caught_ex = None
 
     model_builder_llama_inference_component.sagemaker_session = sagemaker_session
+    model_builder_llama_inference_component.instance_type = INSTANCE_TYPE
 
     model_builder_llama_inference_component.inference_component_name = unique_name_from_base(
         LLAMA_IC_NAME
@@ -88,7 +90,7 @@ def test_model_builder_ic_sagemaker_endpoint(
             logger.info("Deploying and predicting in SAGEMAKER_ENDPOINT mode...")
             endpoint_name = "llama-ic-endpoint-name"
             predictors = chain.deploy(
-                instance_type="ml.g5.24xlarge",
+                instance_type=INSTANCE_TYPE,
                 initial_instance_count=1,
                 accept_eula=True,
                 endpoint_name=endpoint_name,

tests/unit/sagemaker/serve/builder/test_model_builder.py

@@ -2967,7 +2967,8 @@ def test_deploy_existing_inference_component_calls_update_inference_component(
         chain_builder = ModelBuilder(
             modelbuilder_list=[builder1],
         ).build()
-        chain_builder.deploy()
+        inputs = {"endpoint_name": "endpoint-001"}
+        chain_builder.deploy(**inputs)
         assert mock_update_inference_component.call_count == 1
 
     def test_deploy_invalid_inputs(self):