| 443 | |
| 444 | |
| 445 | def parse_arguments(): |
| 446 | args = argparse.ArgumentParser() |
| 447 | args.add_argument("--dataset", type=str, required=True) |
| 448 | args.add_argument("--augmenter", type=str, required=True) |
| 449 | args.add_argument("--output", type=str, required=True) |
| 450 | args = args.parse_args() |
| 451 | |
| 452 | assert args.dataset.endswith(".tsv") or args.dataset.endswith( |
| 453 | ".csv" |
| 454 | ), "Dataset file must be a tsv or csv file, containing a list of files to be augmented" |
| 455 | assert args.output.endswith(".json"), "Output file must be a json file" |
| 456 | |
| 457 | return args |
| 458 | |
| 459 | |
| 460 | def read_data(args): |