(args)
| 150 | |
| 151 | |
| 152 | def main(args): |
| 153 | parser = argparse.ArgumentParser() |
| 154 | parser.add_argument( |
| 155 | "--use-pt-pinned-commit", |
| 156 | action="store_true", |
| 157 | help="build from the pinned PyTorch commit instead of nightly", |
| 158 | ) |
| 159 | parser.add_argument( |
| 160 | "--example", |
| 161 | action="store_true", |
| 162 | help="Also installs required packages for running example scripts.", |
| 163 | ) |
| 164 | args = parser.parse_args(args) |
| 165 | use_pytorch_nightly = not bool(args.use_pt_pinned_commit) |
| 166 | install_requirements(use_pytorch_nightly) |
| 167 | if args.example: |
| 168 | install_optional_example_requirements(use_pytorch_nightly) |
| 169 | |
| 170 | |
| 171 | if __name__ == "__main__": |
no test coverage detected