def METHOD_NAME(self, split, epoch=1, combine=False, **kwargs):
    self.datasets[split] = {
        "data": json.load(open(f"{self.cfg.data}/json/{split}.json")),
        "data_dir": self.cfg.data,
        "shuffle": True if split == "train" else False,
    }
    self.datasets[split] = Namespace(**self.datasets[split])