()
| 575 | |
| 576 | |
| 577 | def parse_args(): |
| 578 | parser = argparse.ArgumentParser() |
| 579 | |
| 580 | parser.add_argument( |
| 581 | "-i", |
| 582 | "--input-file", |
| 583 | type=str, |
| 584 | default="python/paddle/tensor/tensor.prototype.pyi", |
| 585 | ) |
| 586 | parser.add_argument( |
| 587 | "-o", |
| 588 | "--output-file", |
| 589 | type=str, |
| 590 | default="python/paddle/tensor/tensor.pyi", |
| 591 | ) |
| 592 | |
| 593 | args = parser.parse_args() |
| 594 | |
| 595 | return args |
| 596 | |
| 597 | |
| 598 | def generate_stub_file(input_file=None, output_file=None): |