Jenkins fix

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"
             }
         }
     }