Skip to content

Commit cd4601e

Browse files
authored
Merge pull request #25 from zenml-io/feature/adjust-to-better-metadata
Remove .value from metadata access
2 parents 0fba996 + a3955d0 commit cd4601e

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

.gitignore

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -161,4 +161,5 @@ cython_debug/
161161

162162
.vscode/
163163
.local/
164-
.ruff_cache/
164+
.ruff_cache/
165+
.idea/

template/steps/promotion/{% if metric_compare_promotion %}promote_get_metrics.py{% endif %}

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def promote_get_metrics(
3636

3737
# Get current model version metric in current run
3838
model = get_step_context().model
39-
current_metrics = model.get_model_artifact("model").run_metadata["metrics"].value
39+
current_metrics = model.get_model_artifact("model").run_metadata["metrics"]
4040
logger.info(f"Current model version metrics are {current_metrics}")
4141

4242
# Get latest saved model version metric in target environment
@@ -48,7 +48,7 @@ def promote_get_metrics(
4848
except KeyError:
4949
latest_version = None
5050
if latest_version:
51-
latest_metrics = latest_version.get_model_artifact("model").run_metadata["metrics"].value
51+
latest_metrics = latest_version.get_model_artifact("model").run_metadata["metrics"]
5252
logger.info(f"Latest model version metrics are {latest_metrics}")
5353
else:
5454
logger.info("No currently promoted model version found.")

0 commit comments

Comments
 (0)