setup.py

"""Setup."""
from pybind11.setup_helpers import Pybind11Extension
from setuptools import find_namespace_packages
from setuptools import setup

__version__ = "0.2.4"

ext_modules = [
    Pybind11Extension(
        "pygenstability.generalized_louvain",
        ["src/pygenstability/generalized_louvain/generalized_louvain.cpp"],
        include_dirs=["extra", "generalizedLouvain"],
        extra_compile_args=["-std=c++11"],
        optional=True,
    ),
]
plotly_require = ["plotly>=3.6.0"]

test_require = [
    "pyyaml",
    "dictdiffer",
    "pytest",
    "pytest-cov",
    "pytest-html",
    "diff-pdf-visually",
    "ipython!=8.7.0",  # see https://github.com/spatialaudio/nbsphinx/issues/687
]
install_requires = [
    "numpy>=1.18.1",
    "scipy>=1.4.1",
    "matplotlib>=3.1.3",
    "scikit-learn",
    "cmake>=3.16.3",
    "click>=7.0",
    "tqdm>=4.45.0",
    "pybind11>=2.10.0",
    "pandas>=1.0.0",
    "threadpoolctl",
]
leiden_install = ["igraph", "leidenalg"]
networkx_install = ["networkx>=3.0"]

setup(
    name="PyGenStability",
    version=__version__,
    author="Alexis Arnaudon, Dominik Schindler",
    author_email="alexis.arnaudon@epfl.ch, dominik.schindler19@imperial.ac.uk",
    url="https://github.com/ImperialCollegeLondon/PyGenStability",
    description="Python binding of generalised Louvain with Markov Stability",
    long_description=open("README.md", "r").read(),
    long_description_content_type="text/markdown",
    ext_modules=ext_modules,
    setup_requires=["pybind11>=2.10.0"],
    install_requires=install_requires,
    zip_safe=False,
    extras_require={
        "plotly": plotly_require,
        "leiden": leiden_install,
        "networkx": networkx_install,
        "all": plotly_require + test_require + leiden_install + networkx_install,
    },
    entry_points={"console_scripts": ["pygenstability=pygenstability.app:cli"]},
    packages=find_namespace_packages("src"),
    include_package_data=True,
    package_dir={"": "src"},
)