Revert file parameter

This commit is contained in:
Marcin Kostrzewski 2022-05-11 20:47:50 +02:00
parent 36e6ac8270
commit e9d1153a35
2 changed files with 1 additions and 6 deletions

View File

@ -5,10 +5,6 @@ pipeline {
} }
} }
parameters { parameters {
file(
name: 'input_example.json',
description: 'Example input file for the model'
)
buildSelector( buildSelector(
defaultSelector: lastSuccessful(), defaultSelector: lastSuccessful(),
description: 'Which build to use for copying artifacts', description: 'Which build to use for copying artifacts',
@ -24,7 +20,6 @@ pipeline {
} }
stage('Predict values using model from artifact') { stage('Predict values using model from artifact') {
steps { steps {
sh "cat input_example.json"
sh "python predict_s444356.py" sh "python predict_s444356.py"
} }
} }

View File

@ -6,7 +6,7 @@ logged_model = 'mlruns/1/296d6f314bb2451885fb7ae58988301e/artifacts/model'
loaded_model = mlflow.pyfunc.load_model(logged_model) loaded_model = mlflow.pyfunc.load_model(logged_model)
with open('input_example.json') as f: with open(f'{logged_model}/input_example.json') as f:
data = json.load(f) data = json.load(f)
input_example = np.array([data['inputs'][0]], dtype=np.float64).reshape(-1, 2) input_example = np.array([data['inputs'][0]], dtype=np.float64).reshape(-1, 2)