addedthe build of the reference map to the dataset

This commit is contained in:
Karma Riuk
2025-05-14 20:52:17 +02:00
parent 5328fe59e1
commit cabf8fd823

View File

@ -189,6 +189,14 @@ class Dataset:
return Dataset(entries=entries) 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__": if __name__ == "__main__":
from utils import prompt_yes_no from utils import prompt_yes_no