Jenkinsfile update

.gitignore

@@ -156,4 +156,6 @@ Car_Prices_Poland_Kaggle*
 CarPrices*
 IUM08/*
 .DS_store
-*.db
+*.db
+mlruns
+my_model

Jenkinsfile_training

@@ -9,9 +9,14 @@ pipeline {
         string(name: 'epoch', defaultValue: '100', description: 'Number of epochs to train model.')
     }
     stages {
-        stage('Get arifacts') {
+        stage('Get artifacts') {
             steps {
                 copyArtifacts fingerprintArtifacts: true, projectName: 's444507-create-dataset', selector: lastSuccessful()
+            }
+        }
+        stage('Run mlflow script and save artifacts') {
+            steps {
+                sh "python3 lab08_deepLearining_mlflow.py $epoch"
                 archiveArtifacts artifacts: 'CarPrices_pytorch_model.pkl'
                 archiveArtifacts artifacts: 'mlruns/**'
                 archiveArtifacts artifacts: 'my_model/**'
@@ -19,11 +24,6 @@ pipeline {
                 sh 'rm -r my_model'
             }
         }
-        stage('Train model with sacred') {
-            steps {
-                sh "python3 lab08_deepLearining_mlflow.py $epoch"
-            }
-        }
     }
     post {
         always {