diff --git a/Dockerfile b/Dockerfile index 6841cc6..3dff01f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -12,10 +12,12 @@ COPY ./clean_and_split_data.py calc_stats.py ./ COPY ./train_model.py ./train_model_mlflow.py ./ COPY ./evaluate.py ./ -RUN apt-get update && apt-get install -y python3-pip unzip && rm -rf /var/lib/apt/lists/* +RUN apt-get update && apt-get install -y python3-pip unzip sshpass openssh-client && rm -rf /var/lib/apt/lists/* RUN export PATH="$PATH:/root/.local/bin" -RUN pip3 install kaggle pandas scikit-learn tensorflow keras matplotlib numpy sacred mlflow +RUN pip3 install kaggle pandas scikit-learn tensorflow keras matplotlib numpy sacred mlflow dvc 'dvc[ssh]' paramiko RUN mkdir /.kaggle && chmod o+w /.kaggle + +RUN useradd -r -u 111 jenkins