add docker file for testing prediction

This commit is contained in:
Michael Duong 2023-09-10 16:11:30 +01:00
parent f76c238ac6
commit fbdf4969a6

View file

@ -0,0 +1,14 @@
# Dockerfile that can be used to test loading a model to generate a prediction (part of CI/CD flow)
FROM python:3.10.12-slim
COPY src/requirements/prediction/requirements-dev.txt requirements.txt
RUN pip install --upgrade pip
RUN pip install -r requirements.txt
# Assuming in the CI/CD step, there will be a dvc pull step to get data and model, so will just need to run a single script
COPY src/ /home/src/
WORKDIR /home/src/
CMD [ "python", "generate_predictions.py"]