from setuptools import setup, find_packages

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