Change the logic from waiting endpoint be inservice, to refresh

Author: mollyheamazon

sagemaker-serve/tests/integ/test_huggingface_integration.py

@@ -101,10 +101,10 @@ def build_and_deploy():
     core_endpoint = model_builder.deploy(endpoint_name=f"{ENDPOINT_NAME_PREFIX}-{unique_id}")
     logger.info(f"Endpoint Successfully Created: {core_endpoint.endpoint_name}")
 
-    # Wait for endpoint to be in service
-    logger.info("Waiting for endpoint to be in service...")
-    core_endpoint.wait_for_status(status="InService")
-    logger.info("Endpoint is now in service")
+    # Verify endpoint exists before proceeding
+    logger.info("Verifying endpoint was created...")
+    core_endpoint.refresh()
+    logger.info(f"Endpoint status: {core_endpoint.endpoint_status}")
 
     return core_model, core_endpoint