From 291a28cc185868083be63a228644910caf318bc1 Mon Sep 17 00:00:00 2001
From: Adam Wojdyla <adam.wojdyla@outlook.com>
Date: Mon, 16 May 2022 03:25:54 +0200
Subject: [PATCH] fix

---
 Jenkinsfile_predict  | 2 +-
 Jenkinsfile_training | 8 +++++---
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/Jenkinsfile_predict b/Jenkinsfile_predict
index 2e71594..128272f 100644
--- a/Jenkinsfile_predict
+++ b/Jenkinsfile_predict
@@ -23,7 +23,7 @@ pipeline {
         }
         stage('Run prediction on model') {
             steps {
-                sh "python3 lab08_predict.py $epoch"
+                sh "python3 lab08_predict.py"
             }
         }
     }
diff --git a/Jenkinsfile_training b/Jenkinsfile_training
index ab0eb5a..a49f273 100644
--- a/Jenkinsfile_training
+++ b/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'
         }