ML/modules/ml-pipeline/src/Prediction.Dockerfile
2023-09-18 20:10:06 +01:00

14 lines
491 B
Docker

# 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 pipeline/requirements/predictions/requirements.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 pipeline/ /home/pipeline/
WORKDIR /home/pipeline/
CMD [ "python", "generate_predictions.py"]