From 24d2eb83cb29d64b404d28fbe96d34434efde107 Mon Sep 17 00:00:00 2001 From: Marcin Kostrzewski Date: Wed, 11 May 2022 20:55:34 +0200 Subject: [PATCH] Add input param --- Jenkinsfile-predict-s444356 | 7 +++++++ predict_s444356.py | 2 +- 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/Jenkinsfile-predict-s444356 b/Jenkinsfile-predict-s444356 index 5a9f0f9..b4b92da 100644 --- a/Jenkinsfile-predict-s444356 +++ b/Jenkinsfile-predict-s444356 @@ -5,6 +5,12 @@ pipeline { } } parameters { + string( + defaultValue: '{"inputs": [[0.51, 0.86], [0.79, 0.79], [0.74, 0.77], [0.66, 0.73]]}', + description: 'Input', + name: 'INPUT', + trim: true + ) buildSelector( defaultSelector: lastSuccessful(), description: 'Which build to use for copying artifacts', @@ -20,6 +26,7 @@ pipeline { } stage('Predict values using model from artifact') { steps { + sh "echo ${params.INPUT} > input_example.json" sh "python predict_s444356.py" } } diff --git a/predict_s444356.py b/predict_s444356.py index 247c1e3..89fb5a1 100644 --- a/predict_s444356.py +++ b/predict_s444356.py @@ -6,7 +6,7 @@ logged_model = 'mlruns/1/296d6f314bb2451885fb7ae58988301e/artifacts/model' loaded_model = mlflow.pyfunc.load_model(logged_model) -with open(f'{logged_model}/input_example.json') as f: +with open('input_example.json') as f: data = json.load(f) input_example = np.array([data['inputs'][0]], dtype=np.float64).reshape(-1, 2)