def lambda_handler()

in api-lambda-sm/lambda-handler.py [0:0]

• 11 lines of code
• 1 McCabe index (conditional complexity)

def lambda_handler(event, context):
    # TODO implement
    
    client = boto3.client('sagemaker-runtime')
    
    print("Calling Sagemaker endpoint")

    response = client.invoke_endpoint(
        EndpointName=os.environ['endpoint_name'],
        Body=json.loads(event['body'])['data'],
        ContentType=os.environ['content_type'])
    
    return {
        'statusCode': 200,
        'body': json.loads(response['Body'].read())
    }