diff --git a/training.Jenkinsfile b/training.Jenkinsfile
index e4eb38b..d8cd737 100644
--- a/training.Jenkinsfile
+++ b/training.Jenkinsfile
@@ -26,7 +26,7 @@ pipeline {
         stage('Training') {
             steps {
             	copyArtifacts filter: '*', projectName:'s444421-create-dataset', selector: buildParameter('BUILD_SELECTOR')
-            	sh 'python training_mlflow.py $EPOCHS'
+            	sh 'ipython ./training_mlflow.py $EPOCHS'
             	archiveArtifacts artifacts: 'mlruns/**'
             }
         }