Skip to content

Commit acb4f47

Browse files
committed
1 parent 4dbef59 commit acb4f47

File tree

3 files changed

+4
-5
lines changed

3 files changed

+4
-5
lines changed

template/quickstart.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -982,8 +982,8 @@
982982
"@pipeline\n",
983983
"def inference(preprocess_pipeline_id: UUID):\n",
984984
" \"\"\"Model batch inference pipeline\"\"\"\n",
985-
" # random_state = client.get_artifact_version(name_id_or_prefix=preprocess_pipeline_id).metadata[\"random_state\"].value\n",
986-
" # target = client.get_artifact_version(name_id_or_prefix=preprocess_pipeline_id).run_metadata['target'].value\n",
985+
" # random_state = client.get_artifact_version(name_id_or_prefix=preprocess_pipeline_id).metadata[\"random_state\"]\n",
986+
" # target = client.get_artifact_version(name_id_or_prefix=preprocess_pipeline_id).run_metadata['target']\n",
987987
" random_state = 42\n",
988988
" target = \"target\"\n",
989989
"\n",

template/run.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -207,8 +207,8 @@ def main(
207207

208208
# Use the metadata of feature engineering pipeline artifact
209209
# to get the random state and target column
210-
random_state = preprocess_pipeline_artifact.run_metadata["random_state"].value
211-
target = preprocess_pipeline_artifact.run_metadata["target"].value
210+
random_state = preprocess_pipeline_artifact.run_metadata["random_state"]
211+
target = preprocess_pipeline_artifact.run_metadata["target"]
212212
run_args_inference["random_state"] = random_state
213213
run_args_inference["target"] = target
214214

template/steps/model_promoter.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,6 @@ def model_promoter(accuracy: float, stage: str = "production") -> bool:
4848
prod_accuracy = (
4949
stage_model.get_artifact("sklearn_classifier")
5050
.run_metadata["test_accuracy"]
51-
.value
5251
)
5352
if float(accuracy) > float(prod_accuracy):
5453
# If current model has better metrics, we promote it

0 commit comments

Comments
 (0)