diff --git a/modules/ml-pipeline/src/pipeline/Prediction.Dockerfile b/modules/ml-pipeline/src/pipeline/Prediction.Dockerfile new file mode 100644 index 0000000..d8e9b6c --- /dev/null +++ b/modules/ml-pipeline/src/pipeline/Prediction.Dockerfile @@ -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"]