()
| 809 | |
| 810 | |
| 811 | def main() -> None: |
| 812 | args = parse_args() |
| 813 | input_dir = Path(args.input_dir) |
| 814 | output_dir = ( |
| 815 | Path(args.output_dir) |
| 816 | if args.output_dir |
| 817 | else input_dir.parent / "main_experiment" / METHOD_KEY |
| 818 | ) |
| 819 | result = run_baseline( |
| 820 | input_dir=input_dir, |
| 821 | output_dir=output_dir, |
| 822 | roles_file=Path(args.roles_file), |
| 823 | top_k=args.top_k, |
| 824 | ) |
| 825 | summary = result["evaluation"]["summary"] |
| 826 | print(json.dumps(summary, ensure_ascii=False, indent=2)) |
| 827 | print(f"\nDiscourse-Aware outputs written to: {output_dir}") |
| 828 | |
| 829 | |
| 830 | if __name__ == "__main__": |
no test coverage detected