setup.py

"""Package setup"""
from setuptools import setup

from gen_pyi import main as gen_pyi

with open("README.md", "r") as fh:
    long_description = fh.read()

# generate pyi files from pyi.in files
gen_pyi()

setup(
    name="data-science-types",
    version="0.3.0.dev1",
    url="https://github.com/predictive-analytics-lab/data-science-types",
    project_urls={
        "Bug Tracker": "https://github.com/predictive-analytics-lab/data-science-types/issues/",
        "Source Code": "https://github.com/predictive-analytics-lab/data-science-types",
    },
    author="PAL",
    description="Type stubs for Python machine learning libraries",
    license="Apache License 2.0",
    long_description=long_description,
    long_description_content_type="text/markdown",
    package_data={
        "matplotlib-stubs": [
            "__init__.pyi",
            "artist.pyi",
            "axes.pyi",
            "backend_bases.pyi",
            "collections.pyi",
            "color.pyi",
            "cm.pyi",
            "font_manager.pyi",
            "image.pyi",
            "legend.pyi",
            "patheffects.pyi",
            "pyplot.pyi",
            "style.pyi",
            "text.pyi",
            "transforms.pyi",
        ],
        "numpy-stubs": ["__init__.pyi", "linalg.pyi", "ma.pyi", "random.pyi", "testing.pyi"],
        "pandas-stubs": ["__init__.pyi", "testing.pyi"],
        "pandas-stubs.core": [
            "__init__.pyi",
            "frame.pyi",
            "indexing.pyi",
            "series.pyi",
            "strings.pyi",
        ],
        "pandas-stubs.core.arrays": ["__init__.pyi", "base.pyi", "integer.pyi", "masked.pyi"],
        "pandas-stubs.core.dtypes": ["__init__.pyi", "base.pyi"],
        "pandas-stubs.core.groupby": ["__init__.pyi", "generic.pyi"],
        "pandas-stubs.core.indexes": ["__init__.pyi", "base.pyi", "frozen.pyi", "multi.pyi"],
    },
    packages=[
        "matplotlib-stubs",
        "numpy-stubs",
        "pandas-stubs",
        "pandas-stubs.core",
        "pandas-stubs.core.arrays",
        "pandas-stubs.core.dtypes",
        "pandas-stubs.core.groupby",
        "pandas-stubs.core.indexes",
    ],
    python_requires=">=3.6",
    # use `pip install data-science-types[dev]` to install development packages
    extras_require={
        "dev": [
            "black",
            "flake8",
            "flake8-pyi",
            "matplotlib",
            "mypy==0.770",
            "numpy",
            "pandas",
            "pytest",
        ]
    },
    classifiers=[  # classifiers can be found here: https://pypi.org/classifiers/
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: Apache Software License",
        "Typing :: Typed",
    ],
    zip_safe=False,
)