From 17645f77a783f8b80b5bd674b531db3287f75b23 Mon Sep 17 00:00:00 2001 From: Samrudhi Sharma Date: Fri, 17 May 2024 10:31:06 -0700 Subject: [PATCH] Integ test updates --- tests/integ/sagemaker/serve/test_serve_tei.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/integ/sagemaker/serve/test_serve_tei.py b/tests/integ/sagemaker/serve/test_serve_tei.py index c3a696d208..29aaa69d7f 100644 --- a/tests/integ/sagemaker/serve/test_serve_tei.py +++ b/tests/integ/sagemaker/serve/test_serve_tei.py @@ -107,8 +107,7 @@ def test_tei_sagemaker_endpoint(sagemaker_session, model_builder, model_input): with timeout(minutes=SERVE_SAGEMAKER_ENDPOINT_TIMEOUT): try: logger.info("Deploying and predicting in SAGEMAKER_ENDPOINT mode...") - predictor = model.deploy(instance_type="ml.g5.12xlarge", initial_instance_count=2) - logger.info("Endpoint successfully deployed.") + predictor = model.deploy(instance_type="ml.g5.2xlarge", initial_instance_count=1) predictor.predict(model_input) assert predictor is not None except Exception as e: