diff --git a/report/evaluation.py b/report/evaluation.py index c54fe7fd833e3890862f34e112690a99ed9a6bcc..4ce2f76cfde8d38f3497e38ee072ffd30e4fe0b1 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}