From cabf8fd8231a6a1f63305a77781e9f8b0488f18d Mon Sep 17 00:00:00 2001 From: Karma Riuk Date: Wed, 14 May 2025 20:52:17 +0200 Subject: [PATCH] addedthe build of the reference map to the dataset --- dataset.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/dataset.py b/dataset.py index f6c0968..ca5b27b 100644 --- a/dataset.py +++ b/dataset.py @@ -189,6 +189,14 @@ class Dataset: return Dataset(entries=entries) + def build_reference_map(self) -> Dict[str, DatasetEntry]: + """Build a reference map for the dataset""" + + ref_map = {} + for entry in self.entries: + ref_map[entry.metadata.id] = entry + return ref_map + if __name__ == "__main__": from utils import prompt_yes_no