test

Dockerfile

@@ -19,4 +19,8 @@ RUN pip3 install --user keras
 RUN pip3 install --user GitPython
 RUN pip3 install --user pymongo
 
+WORKDIR /app
 
+COPY ./train.py ./
+
+CMD ./train.py

Jenkinsfile

@@ -1,5 +1,5 @@
 pipeline {
-    agent any;
+    agent { dockerfile true }
 
    parameters{
       buildSelector(
@@ -31,11 +31,9 @@ pipeline {
             {
                 steps
                 {
-			withDockerContainer('shroomy/ium2:2') {
                        sh 'python3 train.py ${BATCH_SIZE} ${EPOCHS}'
                 }
             }
-            }
 
 
 		stage('Archive artifacts') {