setup.py

# Copyright 2020 Google
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import functools
import operator
import pathlib
import runpy

from setuptools import find_packages, setup

__version__ = runpy.run_path('recirq/_version.py')['__version__']
assert __version__, 'Version string cannot be empty'

description = (
    'A collection of Python modules that implement algorithms and experiments '
    'in quantum computing, written in and using Cirq. Research using Cirq!'
)


def _parse_requirements(path: pathlib.Path):
    lines = [line.strip() for line in path.read_text().splitlines() if line]
    return [line for line in lines if not line.startswith('#')]


install_requires = _parse_requirements(pathlib.Path('requirements.txt'))
requirements_files = list(pathlib.Path('recirq').glob('*/extra-requirements.txt'))
assert requirements_files, 'No recirq/*/extra-requirements.txt files found'
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.
install_requires = functools.reduce(operator.add, extras_require.values(), install_requires)

setup(name='recirq',
      version=__version__,
      url='http://github.com/quantumlib/recirq',
      author='Quantum AI team and collaborators',
      maintainer_email="quantum-oss-maintainers@google.com",
      python_requires='>=3.10.0',
      install_requires=install_requires,
      extras_require=extras_require,
      license='Apache 2',
      description=description,
      long_description=open('README.md', encoding='utf-8').read(),
      long_description_content_type='text/markdown',
      packages=find_packages(),
      package_data={'recirq': [
          # https://github.com/quantumlib/ReCirq/issues/101
          'hfvqe/molecular_data/hydrogen_chains/*/*/*',
          # Needed for ftbbl docs
          'otoc/loschmidt/tilted_square_lattice/*',
          # Needed for qcqmc notebooks
          'qcqmc/data/integrals/*/*',
      ]},
      )