diff --git a/train-eval/Dockerfile b/train-eval/Dockerfile
index e8aa1c8..93783f0 100644
--- a/train-eval/Dockerfile
+++ b/train-eval/Dockerfile
@@ -15,6 +15,7 @@ RUN pip install scikit-learn
 # RUN echo "alias kaggle='~/.local/bin/kaggle'" >> ~/.bashrc
 
 COPY ./learning.py ./
+COPY ./train-eval/learning.py ./train-eval
 
 # CMD ./script.sh 300
 # CMD ./learning.pyRUN pip install --user numpy