fix

Jenkinsfile_predict

@@ -23,7 +23,7 @@ pipeline {
         }
         stage('Run prediction on model') {
             steps {
-                sh "python3 lab08_predict.py $epoch"
+                sh "python3 lab08_predict.py"
             }
         }
     }

Jenkinsfile_training

@@ -9,6 +9,11 @@ pipeline {
         stage('Get arifacts') {
             steps {
                 copyArtifacts fingerprintArtifacts: true, projectName: 's444507-create-dataset', selector: lastSuccessful()
+                archiveArtifacts artifacts: 'CarPrices_pytorch_model.pkl'
+                archiveArtifacts artifacts: 'mlruns/**'
+                archiveArtifacts artifacts: 'my_model/**'
+                sh 'rm -r mlruns'
+			    sh 'rm -r my_model'
             }
         }
         stage('Train model with sacred') {
@@ -18,9 +23,6 @@ pipeline {
         }
     }
     post {
-        success {
-            archiveArtifacts artifacts: 'CarPrices_pytorch_model.pkl, mlruns/**, my_model/**', followSymlinks: false
-        }
         always {
             emailext body: "${currentBuild.currentResult}", subject: 's444507-training', to: 'e19191c5.uam.onmicrosoft.com@emea.teams.ms'
         }