From 54f5e68caf12f0133b909288e7b7f480cfc65724 Mon Sep 17 00:00:00 2001
From: Andrzej Preibisz <andrzejjpreibisz@gmail.com>
Date: Mon, 2 May 2022 21:06:30 +0200
Subject: [PATCH] Jenkins fix

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

diff --git a/Jenkinsfile.train b/Jenkinsfile.train
index 1f4f174..c01a7ff 100644
--- a/Jenkinsfile.train
+++ b/Jenkinsfile.train
@@ -3,7 +3,7 @@ pipeline {
         dockerfile true
     }
     parameters {
-        string(name: 'EPOCH', defaultValue: '10', description: 'Number of epochs in training')
+        string(name: 'EPOCHS', defaultValue: '10', description: 'Number of epochs in training')
     }
     stages {
         stage("Git clone") {
@@ -14,7 +14,7 @@ pipeline {
         stage("Training") {
             steps {
                     sh "chmod 777 ml_training.py"
-                    sh "python3 ml_training.py $EPOCH"
+                    sh "python3 ml_training.py"
             }
         }
     }