(dataset_cfg: ConfigDict)
| 42 | |
| 43 | |
| 44 | def parse_dataset_cfg(dataset_cfg: ConfigDict) -> Dict[str, ConfigDict]: |
| 45 | dataset2cfg = {} |
| 46 | for dataset in dataset_cfg: |
| 47 | dataset2cfg[dataset_abbr_from_cfg(dataset)] = dataset |
| 48 | return dataset2cfg |
| 49 | |
| 50 | |
| 51 | def print_prompts(model_cfg, dataset_cfg, count=1): |
no test coverage detected