def lambda_handler()

in 01-byoc/invoke_endpoint/lambda_function.py [0:0]

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

def lambda_handler(event, context):
    body = event["content"]
    payload = base64.b64decode(body)
    runtime_client = boto3.client('runtime.sagemaker')
    response = runtime_client.invoke_endpoint(EndpointName=ENDPOINT_NAME, 
                                  ContentType='application/octet-stream', 
                                  Body=payload)
    
    result = response['Body'].read().decode('utf-8')    
    return json.loads(result)