def handler()

in tensorflow-train-in-sagemaker-deploy-with-lambda/container/app/app.py [0:0]

• 16 lines of code
• 2 McCabe index (conditional complexity)

def handler(event, context):
    print('Received event: ' + json.dumps(event, indent=2))

    destination = '/tmp/' + event["file"]
    s3.download_file(event["bucket"], event["prefix"] + event["file"], destination)
    data = np.load(destination)

    predictions = infer(tf.constant(data))['dense_1']
    print('predictions: {}'.format(predictions))

    result=[]
    for element in predictions:
        prediction = np.argmax(element)
        result.append(int(prediction))

    print('Returning result: {}'.format(result))

    return {
        'statusCode': 200,
        'body': json.dumps(result)
    }