From db598936a9c1d68b492bb970a120bedcef194b80 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jakub=20Zar=C4=99ba?= <k.mjayk@gmail.com>
Date: Wed, 10 May 2023 16:54:08 +0200
Subject: [PATCH] s

---
 JenkinsfileDL | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

diff --git a/JenkinsfileDL b/JenkinsfileDL
index 3f5a32a..4eab5ab 100644
--- a/JenkinsfileDL
+++ b/JenkinsfileDL
@@ -3,10 +3,7 @@ pipeline {
         docker {
             image 'python:3.11'
             args '-v /root/.cache:/root/.cache'
-            script {
-                // sh 'python3 train.py ${params.EPOCHS}'
-                sh 'pip install pandas && tail -f /dev/null'
-            }
+            command 'sh', '-c', 'pip install pandas tensorflow && tail -f /dev/null'
         }
     }
     parameters {
@@ -23,8 +20,7 @@ pipeline {
         stage('Trenuj model') {
             steps {
                 script {
-                    // sh 'python3 train.py ${params.EPOCHS}'
-                    sh 'python3 train.py'
+                    sh 'python3 train.py --epochs ${params.EPOCHS}'
                 }
             }
         }
@@ -34,4 +30,4 @@ pipeline {
             }
         }
     }
-}
\ No newline at end of file
+}