"""Package setup for the ASCAD training pipeline."""

from setuptools import setup, find_packages

setup(
    name="ascad-training-pipeline",
    version="2.0.0",
    description="Deep learning framework for side-channel analysis on ASCAD",
    packages=find_packages(),
    python_requires=">=3.9",
    install_requires=[
        "tensorflow>=2.13.0",
        "numpy>=1.24.0",
        "h5py>=3.9.0",
        "huggingface_hub>=0.19.0",
        "requests>=2.31.0",
        "tqdm>=4.66.0",
    ],
    extras_require={
        "wandb": ["wandb>=0.16.0"],
        "viz": ["matplotlib>=3.8.0"],
        "orchestrator": [
            "fastapi>=0.110.0",
            "uvicorn[standard]>=0.29.0",
            "click>=8.1.0",
            "pydantic>=2.6.0",
            "pyyaml>=6.0.0",
        ],
    },
    entry_points={
        "console_scripts": [
            "tq=orchestrator.cli.tq:main",
        ],
    },
)