message: Reset method name as this was an accidental commit
diff: @@ -18,8 +18,7 @@ def generate_dataset() -> FederatedDataset:
  return dataset
  -
  -def load():
  +def load_splits():
  return persistence.load_local_dataset(
  dataset_name=DATASET_NAME, local_datasets_dir=get_config("local_datasets_dir")
  )
  