(line)
| 72 | |
| 73 | |
| 74 | def install_package(line): |
| 75 | pat = '(' + '|'.join(['>=', '==', '>', '<', '<=', '@']) + ')' |
| 76 | parts = re.split(pat, line, maxsplit=1) |
| 77 | package_name = parts[0].strip() |
| 78 | print('installing', package_name) |
| 79 | if package_name == 'pytorch3d': |
| 80 | links = pytorch3d_links() |
| 81 | run_subprocess( |
| 82 | [sys.executable, '-m', 'pip', 'install', 'pytorch3d', '-f', links]) |
| 83 | elif package_name == 'mmcv': |
| 84 | links = mmcv_links() |
| 85 | run_subprocess( |
| 86 | [sys.executable, '-m', 'pip', 'install', line, '-f', links]) |
| 87 | elif package_name == 'MinkowskiEngine': |
| 88 | run_subprocess([sys.executable, '-m', 'pip', 'install', 'ninja']) |
| 89 | run_subprocess([ |
| 90 | sys.executable, '-m', 'pip', 'install', '-U', |
| 91 | 'git+https://github.com/NVIDIA/MinkowskiEngine', '--no-deps' |
| 92 | ]) # noqa: E501 |
| 93 | else: |
| 94 | run_subprocess([sys.executable, '-m', 'pip', 'install', line]) |
| 95 | |
| 96 | |
| 97 | def install_requires(fname): |
no test coverage detected