From fbc8395e2c28cfd3fcc940db4453a505ee64c979 Mon Sep 17 00:00:00 2001
From: Andrzej Preibisz <andrzejjpreibisz@gmail.com>
Date: Mon, 2 May 2022 20:34:50 +0200
Subject: [PATCH] Jenkins fix

---
 Jenkinsfile.train | 20 ++++----------------
 1 file changed, 4 insertions(+), 16 deletions(-)

diff --git a/Jenkinsfile.train b/Jenkinsfile.train
index d443e60..1f4f174 100644
--- a/Jenkinsfile.train
+++ b/Jenkinsfile.train
@@ -2,22 +2,10 @@ pipeline {
     agent {
         dockerfile true
     }
+    parameters {
+        string(name: 'EPOCH', defaultValue: '10', description: 'Number of epochs in training')
+    }
     stages {
-        stage("Setup parameters") {
-            steps {
-                script {
-                    properties([
-                        parameters([
-                            string(
-                                defaultValue: '10',
-                                name: 'EPOCHS',
-                                trim: true
-                            )
-                        ])
-                    ])
-                }
-            }
-        }
         stage("Git clone") {
             steps {
               checkout([$class: 'GitSCM', branches: [[name: '*/master']], extensions: [], userRemoteConfigs: [[credentialsId: 's444465', url: 'https://git.wmi.amu.edu.pl/s444465/ium_444465']]])
@@ -26,7 +14,7 @@ pipeline {
         stage("Training") {
             steps {
                     sh "chmod 777 ml_training.py"
-                    sh "python3 ml_training.py"
+                    sh "python3 ml_training.py $EPOCH"
             }
         }
     }