()
| 18 | |
| 19 | |
| 20 | def main(): |
| 21 | args = parse_args() |
| 22 | models = match_files('opencompass/configs/models/', |
| 23 | args.pattern, |
| 24 | fuzzy=True) |
| 25 | if models: |
| 26 | table = [['Model', 'Config Path'], *models] |
| 27 | print(tabulate.tabulate(table, headers='firstrow', tablefmt='psql')) |
| 28 | datasets = match_files('opencompass/configs/datasets/', |
| 29 | args.pattern, |
| 30 | fuzzy=True) |
| 31 | if datasets: |
| 32 | table = [['Dataset', 'Config Path'], *datasets] |
| 33 | print(tabulate.tabulate(table, headers='firstrow', tablefmt='psql')) |
| 34 | |
| 35 | |
| 36 | if __name__ == '__main__': |
no test coverage detected