AWS Lambda using Custom Python Image

The custom container includes

• Python
• Mlflow
• A simple Mlflow model

When deployed to AWS Lambda, the container will add a small integer to any input dataframe and make a json response.

Commands

Train, build and run a container

make train build run

Test the container

make test

Test data

{
    "columns": ["a", "b", "c"],
    "data": [[1, 2, 3], [4, 5, 6]]
}

References

• Mlflow sample AddN model
• AWS Lambda Developer Guide: Create a Python image from an alternative base image
• AWS Blog: Container image support for Lambda