Automatically discover package requirements from extra-requirements.txt files

MANIFEST.in

@@ -1,2 +1,3 @@
 include requirements.txt
 include LICENSE
+recursive-include recirq extra-requirements.txt

dev_tools/write-ci-requirements.py

@@ -74,13 +74,10 @@ def main(*, out_fn: str = 'ci-requirements.txt', relative_cirq_version: str = 'c
     core_reqs = _parse_requirements(REPO_DIR / 'requirements.txt')
     core_reqs = _set_cirq_version(core_reqs, relative_cirq_version)
 
-    extras_require = [
-        'otoc', 'qaoa', 'optimize', 'hfvqe', 'fermi_hubbard', 'qml_lfe',
-        'seniority_zero', 'qcqmc', 'contextuality', 'qec_memory'
-    ]
+    requirements_files = (REPO_DIR / 'recirq').glob('*/extra-requirements.txt')
     extras_require = {
-        r: _parse_requirements(pathlib.Path(REPO_DIR / f'recirq/{r}/extra-requirements.txt'))
-        for r in extras_require
+        req_file.parent.name: _parse_requirements(req_file)
+        for req_file in sorted(requirements_files)
     }
     extras_require['qaoa'] = _set_qaoa_hacks(extras_require['qaoa'], relative_cirq_version)
 

setup.py

@@ -34,13 +34,11 @@ def _parse_requirements(path: pathlib.Path):
 
 
 install_requires = _parse_requirements(pathlib.Path('requirements.txt'))
-extras_require = [
-    'otoc', 'qaoa', 'optimize', 'hfvqe', 'fermi_hubbard', 'qml_lfe',
-    'seniority_zero', 'qcqmc', 'qec_memory'
-]
+requirements_files = list(pathlib.Path('recirq').glob('*/extra-requirements.txt'))
+assert requirements_files, 'No recirq/*/extra-requirements.txt files found'
 extras_require = {
-    r: _parse_requirements(pathlib.Path(f'recirq/{r}/extra-requirements.txt'))
-    for r in extras_require
+    req_file.parent.name: _parse_requirements(req_file)
+    for req_file in sorted(requirements_files)
 }
 
 # TODO(gh-231): remove and require users to install via extras_require.