setup.py

from io import open
from os import path

from setuptools import find_packages, setup

here = path.abspath(path.dirname(__file__))

# Get the long description from the README file
with open(path.join(here, "README.md"), encoding="utf-8") as f:
    long_description = f.read()

setup(
    name="tiktorch",
    version="24.12.0",
    description="Tiktorch client/server",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/ilastik/tiktorch",
    author="ilastik Team",
    author_email="team@ilastik.org",
    classifiers=[  # Optional
        "Development Status :: 3 - Alpha",
        "Intended Audience :: Developers",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: 3.11",
    ],
    packages=find_packages(exclude=["tests"]),  # Required
    install_requires=[
        "bioimageio.spec==0.5.3.3",
        "bioimageio.core==0.6.10",
        "grpcio>=1.31",
        "numpy<2",  # pytorch 2.2.2-py3.9_0 for macos is compiled with numpy 1.*
        "protobuf",
        "pydantic>=2.7.0,<2.10",
        "pytorch3dunet",
        "pyyaml",
        "xarray",
    ],
    extras_require={"server-pytorch": ["pytorch>=2", "scikit-learn", "torchvision"]},
    entry_points={"console_scripts": ["tiktorch=tiktorch.server.base:main"]},
    # extras_require={"test": ["pytest"]},
    project_urls={  # Optional
        "Bug Reports": "https://github.com/ilastik/tiktorch/issues",
        "Source": "https://github.com/ilastik/tiktorch/",
    },
)