Add wait_for_status for flaky test for sagemaker-serve

Author: mollyheamazon

sagemaker-serve/tests/integ/test_huggingface_integration.py

@@ -101,6 +101,11 @@ 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")
+    
     return core_model, core_endpoint