Fix sagemaker-serve integ tests

Author: papriwal

sagemaker-serve/tests/integ/test_tei_integration.py

@@ -31,9 +31,6 @@
 MODEL_NAME_PREFIX = "tei-test-model"
 ENDPOINT_NAME_PREFIX = "tei-test-endpoint"
 
-# Configuration from backup file
-AWS_REGION = "us-east-2"
-
 
 @pytest.mark.slow_test
 def test_tei_build_deploy_invoke_cleanup():
@@ -81,8 +78,6 @@ def build_and_deploy():
     hf_model_id = MODEL_ID
 
     schema_builder = create_schema_builder()
-    boto_session = boto3.Session(region_name=AWS_REGION)
-    sagemaker_session = Session(boto_session=boto_session)
     unique_id = str(uuid.uuid4())[:8]
 
     compute = Compute(
@@ -94,7 +89,6 @@ def build_and_deploy():
         model=hf_model_id,  # Use HuggingFace model string
         model_server=ModelServer.TEI,
         schema_builder=schema_builder,
-        sagemaker_session=sagemaker_session,
         compute=compute,
     )
 

sagemaker-serve/tests/integ/test_tgi_integration.py

@@ -31,9 +31,6 @@
 MODEL_NAME_PREFIX = "tgi-test-model"
 ENDPOINT_NAME_PREFIX = "tgi-test-endpoint"
 
-# Configuration from backup file
-AWS_REGION = "us-east-2"
-
 
 @pytest.mark.slow_test
 def test_tgi_build_deploy_invoke_cleanup():
@@ -81,8 +78,6 @@ def build_and_deploy():
     hf_model_id = MODEL_ID
 
     schema_builder = create_schema_builder()
-    boto_session = boto3.Session(region_name=AWS_REGION)
-    sagemaker_session = Session(boto_session=boto_session)
     unique_id = str(uuid.uuid4())[:8]
 
     compute = Compute(
@@ -101,7 +96,6 @@ def build_and_deploy():
         model=hf_model_id,  # Use HuggingFace model string
         model_server=ModelServer.TGI,
         schema_builder=schema_builder,
-        sagemaker_session=sagemaker_session,
         compute=compute,
         env_vars=env_vars
     )