next fix :(

Jenkinsfile_train

@@ -1,22 +1,38 @@
 pipeline {
-    agent {dockerfile true}
+    agent {docker { image 'snowycocoon/ium_434788:2' }}
-	parameters {
+	parameters{
         buildSelector(
             defaultSelector: lastSuccessful(),
             description: 'Which build to use for copying artifacts',
-					name: 'BUILD_SELECTOR')
+            name: 'WHICH_BUILD'
+        )
         string(
-            defaultValue: '500',
+            defaultValue: '16',
-            description: 'Enter the number of Epochs',
+            description: 'batch size',
-            name: 'epochs',
+            name: 'BATCH_SIZE'
-            trim: false)
+        )
+        string(
+            defaultValue: '15', 
+            description: 'epochs',
+            name: 'EPOCHS'
+        )
     }
 
     stages {
-		stage('copyArtifacts') {
+		stage('copy artifacts')
-            steps {
+            {
-                copyArtifacts fingerprintArtifacts: true, projectName: 's434788-create-dataset', selector: buildParameter('BUILD_SELECTOR')
+                steps
-				sh 'python3 Zadanie_06_training.py ${epochs}'
+                {
+                    copyArtifacts(fingerprintArtifacts: true, projectName: 's434788-create-dataset', selector: buildParameter('WHICH_BUILD'))
+                }
+            }
+            stage('train')
+            {
+                steps
+                {
+                    catchError {
+                        sh 'python3.8 Zadanie_06_training'
+                    }
                 }
             }