setup.py

import re
import ast
from setuptools import setup, find_packages
from glob import glob
from os.path import splitext
from os.path import basename

_version_re = re.compile(r"__version__\s+=\s+(.*)")

with open("src/sklearn_evaluation/__init__.py", "rb") as f:
    VERSION = str(
        ast.literal_eval(_version_re.search(f.read().decode("utf-8")).group(1))
    )

DOWNLOAD_URL = "https://github.com/ploomber/sklearn-evaluation/tarball/{}".format(
    VERSION
)


DEV = [
    "pkgmt",
    "jupytext",
    "ploomber-engine",
    "ipykernel",
    "pytest<8",
    # need to pin this version because pytest 4 breaks matplotlib image
    # comparison tests
    "pytest-cov",
    # TODO: update config so coveralls 3 works
    "coveralls<3",
    # altair is needed for interactive confusion matrix
    "altair",
    # for pandas.read_html (optional dependency for NotebookCollection)
    "lxml",
    # tmp fix, 3.8 breaks the tests:
    # ERROR tests/test_grid_search.py - ImportError: cannot import name 'cleanup'
    # from 'matplotlib.testing.decorators'
    "matplotlib<3.8",
]


setup(
    name="sklearn-evaluation",
    packages=find_packages("src"),
    package_dir={"": "src"},
    py_modules=[splitext(basename(path))[0] for path in glob("src/*.py")],
    version=VERSION,
    description=(
        "scikit-learn model evaluation made easy: plots, tables and" "markdown reports."
    ),
    url="http://github.com/ploomber/sklearn-evaluation",
    download_url=DOWNLOAD_URL,
    author="Ploomber",
    author_email="contact@ploomber.io",
    license="Apache 2",
    keywords=["datascience", "machinelearning"],
    classifiers=[],
    include_package_data=True,
    install_requires=[
        "ploomber-core>=0.2.6",
        "ploomber-extension",
        # compute metrics
        "scikit-learn",
        # plotting
        "matplotlib",
        # misc
        "decorator",
        # metric tables
        "tabulate",
        "jinja2",
        # reports
        "mistune",
        "pandas",
        "nbformat",
        # notebook compare
        "ipython",
        # notebook collection uses this for formatting
        "black",
        # extracting injected parameters from notebooks
        "parso",
        'importlib-metadata;python_version<"3.8"',
    ],
    extras_require={
        "dev": DEV,
    },
)