From 9cc4a521c5351b263a4cf523c644fca4d15575cd Mon Sep 17 00:00:00 2001 From: Kris Thielemans Date: Tue, 16 Jul 2024 17:04:49 +0100 Subject: [PATCH] more pythonic implementation to get keys() Co-authored-by: Casper da Costa-Luis --- petric.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/petric.py b/petric.py index 19997da..2fea399 100755 --- a/petric.py +++ b/petric.py @@ -136,10 +136,7 @@ def evaluate(self, test_im: STIR.ImageData) -> dict[str, float]: return {**whole, **local} def keys(self): - l = ["RMSE_whole_object", "RMSE_background"] - for voi_name in sorted(self.voi_indices.keys()): - l.append(f"AEM_VOI_{voi_name}") - return l + return ["RMSE_whole_object", "RMSE_background"] + [f"AEM_VOI_{name}" for name in sorted(self.voi_indices)] class MetricsWithTimeout(cil_callbacks.Callback):