Merge pull request #46 from zenml-io/bugfix/model-fetching

Fetch artifact from model version without type
zenml-io · Nov 13, 2024 · f67756a · f67756a
2 parents 8570197 + 7c38f0a
commit f67756a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.../steps/{% if hyperparameters_tuning %}hp_tuning{% endif %}/hp_tuning_select_best_model.py b/.../steps/{% if hyperparameters_tuning %}hp_tuning{% endif %}/hp_tuning_select_best_model.py
@@ -31,7 +31,7 @@ def hp_tuning_select_best_model(
     best_metric = -1
     # consume artifacts attached to current model version in Model Control Plane
     for step_name in step_names:
-        hp_output = model.get_data_artifact("hp_result")
+        hp_output = model.get_artifact("hp_result")
         model_: ClassifierMixin = hp_output.load()
         # fetch metadata we attached earlier
         metric = float(hp_output.run_metadata["metric"])