Skip to content

Commit f67756a

Browse files
authored
Merge pull request #46 from zenml-io/bugfix/model-fetching
Fetch artifact from model version without type
2 parents 8570197 + 7c38f0a commit f67756a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

template/steps/{% if hyperparameters_tuning %}hp_tuning{% endif %}/hp_tuning_select_best_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def hp_tuning_select_best_model(
3131
best_metric = -1
3232
# consume artifacts attached to current model version in Model Control Plane
3333
for step_name in step_names:
34-
hp_output = model.get_data_artifact("hp_result")
34+
hp_output = model.get_artifact("hp_result")
3535
model_: ClassifierMixin = hp_output.load()
3636
# fetch metadata we attached earlier
3737
metric = float(hp_output.run_metadata["metric"])

0 commit comments

Comments
 (0)