From 5a7e92fd4dec1a7b0560ef59fce7434e62fa6932 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Szymon=20Parafin=CC=81ski?= Date: Mon, 16 May 2022 02:40:30 +0200 Subject: [PATCH] fix dllib.py call --- Jenkinsfile_train | 4 ++-- biblioteka_DL/dllib.py | 3 +-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/Jenkinsfile_train b/Jenkinsfile_train index 6ab4528..2e36c3c 100644 --- a/Jenkinsfile_train +++ b/Jenkinsfile_train @@ -21,8 +21,8 @@ pipeline { } stage('Train model with sacred') { steps { - sh 'python3 ./biblioteka_DL/dllib.py with "epochs=$EPOCHS"' - archiveArtifacts artifacts: 'model.pkl, s444018_sacred_FileObserver/**/*.*, result.csv', followSymlinks: false + sh 'python3 ./biblioteka_DL/dllib.py "epochs=$EPOCHS"' + archiveArtifacts artifacts: 'model.pkl, result.csv', followSymlinks: false archiveArtifacts artifacts: 'mlruns/**' archiveArtifacts artifacts: 'my_model/**' build job: 's444018-evaluation/master/' diff --git a/biblioteka_DL/dllib.py b/biblioteka_DL/dllib.py index 5b6ef99..d61a0c9 100644 --- a/biblioteka_DL/dllib.py +++ b/biblioteka_DL/dllib.py @@ -15,8 +15,7 @@ from urllib.parse import urlparse # mlflow.set_tracking_uri("http://172.17.0.1:5000") mlflow.set_experiment("s444018") - -epochs = int(sys.argv[1]) if len(sys.argv) > 1 else 20 +epochs = sys.argv[1] def drop_relevant_columns(imbd_data):