in backend/training-pipeline/functions/step_functions/check_deployment_status.py [0:0]
def lambda_handler(event, context):
endpoint_arn = event['EndpointArn']
print(f"endpoint_arn = {endpoint_arn}")
endpoint_name = get_endpoint_name(endpoint_arn)
print(f"endpoint_name = {endpoint_name}")
response = client.describe_endpoint(
EndpointName=endpoint_name
)
print(f'Endpoint status is {response["EndpointStatus"]}')
return response['EndpointStatus'] == 'InService'