setup.py

"""judo package installation metadata."""
from importlib.machinery import SourceFileLoader
from pathlib import Path

from setuptools import find_packages, setup


version = SourceFileLoader(
    "judo.version",
    str(Path(__file__).parent / "judo" / "version.py"),
).load_module()

with open(Path(__file__).with_name("README.md"), encoding="utf-8") as f:
    long_description = f.read()

# Module-specific dependencies.
extras = {
    "torch": ["torch>=1.0.0"],
    "image": ["pillow>=7.0.0"],
    "ipython": ["ipython >= 7.0.0"],
    "data-structures": ["networkx > 2.0.0"],
}

# Meta dependency groups.
extras["all"] = [item for group in extras.values() for item in group]

setup(
    name="judo",
    description="API and data structures for efficient AI research.",
    long_description=long_description,
    long_description_content_type="text/markdown",
    packages=find_packages(),
    version=version.__version__,
    license="MIT",
    author="FragileTech",
    author_email="info@fragile.tech",
    url="https://github.com/FragileTech/judo",
    keywords=["Machine learning", "artificial intelligence"],
    test_suite="tests",
    tests_require=["pytest>=5.3.5", "hypothesis>=5.6.0"],
    extras_require=extras,
    install_requires=[
        "numpy>=1.0.0",
        "pyyaml>=5.0.0",
        "xxhash>=1.1.0",
    ],
    package_data={"": ["README.md"], "judo": ["config.yml"]},
    classifiers=[
        "Development Status :: 3 - Alpha",
        "Intended Audience :: Developers",
        "License :: OSI Approved :: MIT License",
        "Programming Language :: Python :: 3.6",
        "Programming Language :: Python :: 3.7",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Topic :: Software Development :: Libraries",
    ],
)