From f6465a49f6b7f21edca843cdf9ea5ea18fa82402 Mon Sep 17 00:00:00 2001 From: Dawid Date: Sun, 23 May 2021 18:10:53 +0200 Subject: [PATCH] Fix 3 --- Jenkisfile_train | 2 +- mlflow_model.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/Jenkisfile_train b/Jenkisfile_train index 5c43511..879942b 100644 --- a/Jenkisfile_train +++ b/Jenkisfile_train @@ -27,7 +27,7 @@ pipeline { stage('archiveArtifacts') { steps{ archiveArtifacts 'vaccines_model/saved_model.pb' - archiveArtifacts 'country_vaccination/**' + archiveArtifacts 'country_vaccination2/**' } } stage('sendMail') { diff --git a/mlflow_model.py b/mlflow_model.py index ec39dd8..e989f76 100644 --- a/mlflow_model.py +++ b/mlflow_model.py @@ -57,8 +57,10 @@ if __name__ == "__main__": mlf.set_experiment("s434804") tracking_url_type_store = urlparse(mlf.get_tracking_uri()).scheme if tracking_url_type_store != "file": + print("no file") mlf.keras.log_model(model, "country_vaccination2", registered_model_name="s434804", signature=signature, input_example=input_example) else: + print("file") mlf.keras.log_model(model, "vaccines_model", signature=signature, input_example=input_example) mlf.keras.save_model(model, "country_vaccination", signature=signature, input_example=input_example) \ No newline at end of file