From eaf64bafa6d451d049b840a5b3d43792db67fe9a Mon Sep 17 00:00:00 2001
From: Jan Nowak <jannow2@st.amu.edu.pl>
Date: Sat, 22 May 2021 23:39:09 +0200
Subject: [PATCH] Dockerfile and Jenkinsfile_train added MLmodel script
 generation.

---
 Dockerfile        | 2 ++
 Jenkinsfile_train | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index da19e61..090a36d 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -33,3 +33,5 @@ COPY ./plot.py ./
 RUN chmod +x plot.py
 COPY ./train_mlflow.py ./
 RUN chmod +x train_mlflow.py
+COPY ./generate_MLmodel.py ./
+RUN chmod +x generate_MLmodel.py
diff --git a/Jenkinsfile_train b/Jenkinsfile_train
index 80d600b..10ed336 100644
--- a/Jenkinsfile_train
+++ b/Jenkinsfile_train
@@ -34,6 +34,8 @@ pipeline {
                         sh 'chmod +x dlgssdpytorch.py'
                         sh 'python3 ./dlgssdpytorch.py $PARAMETRY'
                         sh 'chmod +x train_mlflow.py'
+						sh 'chmod +x generate_MLmodel.py'
+						sh 'python3 ./generate_MLmodel'
                         //sh 'mlflow ./train_mlflow.py'
 						//sh 'mlflow run .'
                     }