def lambda_handler()

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'