NNHandler/setup.py at master · rouzib/NNHandler · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
"""
Setup script for NNHandler package.
"""
from setuptools import setup, find_packages
import os

version_ns = {}
with open(os.path.join("src", "nn_handler", "__version__.py")) as f:
    exec(f.read(), version_ns)
__version__ = version_ns["__version__"]

# Try to read README.rst first (for PyPI), fall back to README.md
if os.path.exists("README.rst"):
    with open("README.rst", "r", encoding="utf-8") as fh:
        long_description = fh.read()
        long_description_content_type = "text/x-rst"
else:
    with open("README.md", "r", encoding="utf-8") as fh:
        long_description = fh.read()
        long_description_content_type = "text/markdown"

setup(
    name="NNHandler",
    version=__version__,
    author="Nicolas Payot",
    author_email="",
    description="A comprehensive framework for PyTorch neural network handling",
    long_description=long_description,
    long_description_content_type=long_description_content_type,
    url="https://github.com/rouzib/NNHandler",
    packages=find_packages(where="src"),
    entry_points={
            "console_scripts": [
                "nn_handler_run = nn_handler.cli:main",  # <== this is new
            ],
        },
    package_dir={"": "src"},
    classifiers=[
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.10",
        "License :: OSI Approved :: MIT License",
        "Operating System :: OS Independent",
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
        "Framework :: Pytorch",
    ],
    python_requires=">=3.10",
    install_requires=[
        "torch>=2.0",
        "cloudpickle>=3.0.0",
        "numpy",
        "matplotlib",
        "tqdm",
        "torch_ema>=0.3",
        "einops"
    ],
)