| 41 | |
| 42 | |
| 43 | def parse_args(): |
| 44 | parser = argparse.ArgumentParser() |
| 45 | |
| 46 | parser.add_argument("--task", default = 'net_delay', type=str, help = 'only support net delay for now' ) |
| 47 | parser.add_argument("--data_path", default = './post_route', type=str, help = 'path to the parent dir of features: nodes, net_edges, pin_positions') |
| 48 | parser.add_argument("--save_path", default = './graph', type=str, help = 'path to save') |
| 49 | |
| 50 | args = parser.parse_args() |
| 51 | return args |
| 52 | |
| 53 | if __name__ == '__main__': |
| 54 | args = parse_args() |