From 3d725fc673093115b718d744532f8b2d33238f70 Mon Sep 17 00:00:00 2001
From: Jozef Kraus <524688@mail.muni.cz>
Date: Fri, 14 Mar 2025 09:47:04 +0100
Subject: [PATCH] fix: update artifact file and key name in
`MlflowEvaluationRetriever` to align with `README.md` from
`fix/summer-school`
---
report/evaluation.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/report/evaluation.py b/report/evaluation.py
index c54fe7f..4ce2f76 100644
--- a/report/evaluation.py
+++ b/report/evaluation.py
@@ -22,9 +22,9 @@ class MlflowEvaluationRetriever(MlflowRetriever):
run_id = run_object.info.run_id
experiment_id = run_object.info.experiment_id
config = self.config[run_id]
- table = self.client.load_table(experiment_id, config["path"], run_ids=[run_id])
- evaluation = polars.from_dataframe(table.astype({config["key"]: "str"})).rename(
- {config["key"]: self.item_key_name}
+ table = self.client.load_table(experiment_id, config["artifact_file"], run_ids=[run_id])
+ evaluation = polars.from_dataframe(table.astype({config["slide_item_key"]: "str"})).rename(
+ {config["slide_item_key"]: self.item_key_name}
)
return {"evaluation": evaluation}
--
GitLab