def lambda_handler()

in online-machine-learning-aws-lambda/app/lambda_inference/app.py [0:0]

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

def lambda_handler(event, _):
    event = APIGatewayProxyEvent(event)
    logger.info(event.__dict__)
    # parse input event
    data = event.get('body')
    data = json.loads(data)
    data = data.get("data")
    # download current model from s3
    regr = _download_model_from_s3(model_name)
    # make prediction
    pred = regr.predict(data)
    # log prediction
    logger.info({
        "data": data,
        "prediction": pred,
    })

    return {
        'statusCode': 200,
        'body': json.dumps(
            {
                "prediction": json.dumps(pred.tolist()),
            }
        )
    }