diff --git a/mu_map/dataset/patches.py b/mu_map/dataset/patches.py index ab8d81fff4b26139779d30ef1613bb9b9aa760a9..248a0a51832880c4cbd0e7a14a067023e7635f5c 100644 --- a/mu_map/dataset/patches.py +++ b/mu_map/dataset/patches.py @@ -45,7 +45,7 @@ class MuMapPatchDataset(MuMapDataset): self.patches = [] self.generate_patches() - def split_copy(self, split_name: str): + def copy(self, split_name: str): return MuMapPatchDataset( dataset_dir=self.dir, patches_per_image=self.patches_per_image,