(BABEL_DIR)
| 74 | |
| 75 | |
| 76 | def load_babel(BABEL_DIR): |
| 77 | print(f"Loading BABEL dataset from: {BABEL_DIR}") |
| 78 | babel_all = {} |
| 79 | for split in BABEL_SPLIT: |
| 80 | json_path = os.path.join(BABEL_DIR, f"{split}.json") |
| 81 | babel_all[split] = process_babel_json(json_path) |
| 82 | return babel_all |
| 83 | |
| 84 | |
| 85 | def load_amass(amass_dir): |
no test coverage detected