def METHOD_NAME(self, fold):
    # load up the ontology
    fn = os.path.join(self.dpath, "apis.json")
    with PathManager.open(fn, "r") as f:
        api_schema_raw = json.load(f)
    chunks = []
    for section in range(20):
        section = f"{section:02d}"
        with PathManager.open(
            os.path.join(self.dpath, f"data_{section}.json")
        ) as f:
            subset = pd.read_json(f)
        chunks.append(subset)
    chunks = pd.concat(chunks, axis=0)
    # deterministic shuffle data for splits
    chunks = chunks.sample(frac=1.0, random_state=42)
    split_size = len(chunks) // 10
    if fold == "train":
        chunks = chunks[: split_size * 8]
    elif fold == "valid":
        chunks = chunks[split_size * 8 : split_size * 9]
    elif fold == "test":
        chunks = chunks[split_size * 9 :]
    return chunks, api_schema_raw