Add parameters to training.py

ML/Jenkinsfile

@@ -5,14 +5,13 @@ pipeline {
             name: 'BUILD_SELECTOR',
             defaultSelector: lastSuccessful(),
             description: 'A build to take the artifacts from'
-        )
+        ),
         string(
             name: 'EPOCHS',
             description: 'Number of epochs',
             defaultValue: '10'
         )
     }
-    }
     stages {
         stage('Copy artifacts') {
             steps {