def lambda_handler()

in scripts/Evaluate/lambda_function.py [0:0]

• 18 lines of code
• 3 McCabe index (conditional complexity)

def lambda_handler(event, context):
    logger.info("Event: {0}".format(event))

    if event['script'] == 'return int(1)':
        return 1
    else: 
        content_type = "text/csv"
        accept = "text/csv"
        endpoint_name = event['script'] # Your endpoint name.
        logger.info("Endpoint: {0}".format(endpoint_name))
        event_data = event['data']
        logger.info("Event data: {0}".format(event_data))
    
        try:
            sagemaker_body = create_sagemaker_body(event_data)
            sagemaker_response = get_sagemaker_response(endpoint_name, content_type, accept, sagemaker_body)
            sagemaker_response_serialized = serialize_sagemaker_response(sagemaker_response)
            return sagemaker_response_serialized
        except Exception as e:
            logger.error(e, exc_info=True)