From f8addf09d4b87aab4b49a716ba29c82fc71015b8 Mon Sep 17 00:00:00 2001
From: s434695 <tomgrz3@st.amu.edu.pl>
Date: Sun, 23 May 2021 19:56:42 +0200
Subject: [PATCH] change for Domino2

---
 Jenkinsfile       | 2 +-
 vgsales-mlflow.py | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/Jenkinsfile b/Jenkinsfile
index f5feaad..24ec09d 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -33,7 +33,7 @@ pipeline {
                 {
                        sh 'python3 train.py ${BATCH_SIZE} ${EPOCHS}'
                        sh 'rm -r my_model'
-                       sh 'python3 vgsales-mlflow.py ${BATCH_SIZE} ${EPOCHS}'
+                       sh 'python3 vgsales-mlflow.py'
                 }
             }
 
diff --git a/vgsales-mlflow.py b/vgsales-mlflow.py
index 57add00..81716fb 100644
--- a/vgsales-mlflow.py
+++ b/vgsales-mlflow.py
@@ -66,4 +66,3 @@ with mlflow.start_run():
     mlflow.log_metric("rmse", rmse) 
     #mlflow.keras.log_model(model, 'vgsales_model.h5')
     mlflow.keras.save_model(model, "my_model", signature=mlflow.models.signature.infer_signature(x_train, y_train), input_example=x_train)
-    
\ No newline at end of file