List of things to be done is the base image correct or we can use a lighter image mlflow or mlflow extras and need to be versioned