()
| 358 | return self.output_file_path |
| 359 | |
| 360 | def arg_parse(): |
| 361 | import argparse |
| 362 | parser = argparse.ArgumentParser() |
| 363 | parser.add_argument("--dataset", type=str, default='example', help="The name of dataset") |
| 364 | parser.add_argument("--query_file", type=str, default='rag_dataset.json', help="The name of anno_file") |
| 365 | parser.add_argument("--experiment_type", type=str, default='cot', help="The type of experiment") |
| 366 | parser.add_argument("--workers_num", type=int, default=10, help="The number of workers") |
| 367 | parser.add_argument("--topk", type=int, default=10, help="The number of topk") |
| 368 | args = parser.parse_args() |
| 369 | return args |
| 370 | |
| 371 | if __name__ == "__main__": |
| 372 | args = arg_parse() |
no outgoing calls
no test coverage detected