diff --git a/modules/ml-pipeline/src/pipeline/Prediction.Dockerfile b/modules/ml-pipeline/src/pipeline/Prediction.Dockerfile index d8e9b6c..096b51b 100644 --- a/modules/ml-pipeline/src/pipeline/Prediction.Dockerfile +++ b/modules/ml-pipeline/src/pipeline/Prediction.Dockerfile @@ -1,7 +1,7 @@ # 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 +COPY src/requirements/predictions/requirements.txt requirements.txt RUN pip install --upgrade pip RUN pip install -r requirements.txt diff --git a/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements-dev.txt b/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements-dev.txt new file mode 100644 index 0000000..28f6af4 --- /dev/null +++ b/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements-dev.txt @@ -0,0 +1,11 @@ +joblib==1.3.2 +boto3==1.28.17 +pandas==1.5.3 +dvc==3.18.0 +dvc-s3==2.23.0 +gto==1.0.4 +scikit-learn==1.3.0 +pyarrow==13.0.0 +pre-commit==3.3.3 +sphinx==7.2.5 +sphinx_rtd_theme==1.3.0 diff --git a/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements.txt b/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements.txt new file mode 100644 index 0000000..c232cb2 --- /dev/null +++ b/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements.txt @@ -0,0 +1,5 @@ +joblib==1.3.2 +boto3==1.28.17 +pandas==1.5.3 +scikit-learn==1.3.0 +pyarrow==13.0.0