from setuptools import setup, find_packages

setup(
    name="aiunn",
    version="0.1.2",
    packages=find_packages(where="src"),
    package_dir={"": "src"},
    install_requires=[
        "torch",
        "aiia",
        "pillow",
        "torchvision",
        "scikit-learn",
    ],
    dependency_links=[
        "git+https://gitea.fabelous.app/Machine-Learning/AIIA.git#egg=aiia"
    ],
    python_requires=">=3.10",
)