From 56cc31edd32c98172c13d57a6b73fe7b73fb1681 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20=C5=81=C4=85czkowski?= <broketmpl@gmail.com>
Date: Fri, 19 Apr 2024 21:27:39 +0200
Subject: [PATCH] IUM_06 - models/Jenkinsfile fix, update
 evaluation/Jenkinsfile

---
 evaluation/Jenkinsfile | 2 +-
 models/Jenkinsfile     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/evaluation/Jenkinsfile b/evaluation/Jenkinsfile
index 261b630..84e584f 100644
--- a/evaluation/Jenkinsfile
+++ b/evaluation/Jenkinsfile
@@ -11,7 +11,7 @@ pipeline {
         )
         buildSelector(
             defaultSelector: lastSuccessful(),
-            description: 'Build to extract artifacts from',
+            description: 'Build to extract model artifact from',
             name: 'BUILD_SELECTOR'
         )
     }
diff --git a/models/Jenkinsfile b/models/Jenkinsfile
index 254b316..67aca6d 100644
--- a/models/Jenkinsfile
+++ b/models/Jenkinsfile
@@ -49,7 +49,7 @@ pipeline {
 
             steps {
                 sh "chmod +x ./create_model.py"
-                sh "python3 ./create_model.py"
+                sh "python3 ./create_model.py ${params.epochs} ${params.learning_rate} ${params.weight_decay}"
                 archiveArtifacts artifacts: 'models/model.pth', onlyIfSuccessful: true
                 build job: 's464863-evaluation/main', wait: false
             }