def METHOD_NAME(self, dataset) -> None:
    for cuis in dataset['ent_cuis']:
        for cui in cuis:
            if cui not in self.label_map:
                self.label_map[cui] = len(self.label_map)