| """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", | |
| ], | |
| }, | |
| ) | |