# AWS Lambda container image for predictive-classification x lambda.
# Uses the official AWS Lambda Python 3.11 base image so the runtime
# implements the Lambda Runtime API. /var/task is the code root.

FROM public.ecr.aws/lambda/python:3.11

COPY requirements.txt ${LAMBDA_TASK_ROOT}
RUN pip install --no-cache-dir -r ${LAMBDA_TASK_ROOT}/requirements.txt

COPY . ${LAMBDA_TASK_ROOT}

RUN python ${LAMBDA_TASK_ROOT}/train_model.py

CMD ["handler.lambda_handler"]
