AbstractGradientTraining/setup.py at main · benjaminstew/AbstractGradientTraining · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
"""Setup for pip package."""

from setuptools import find_packages
from setuptools import setup

setup(
    name="abstract_gradient_training",
    version="0.3",
    description="Abstract gradient training of neural networks",
    url="https://github.com/psosnin/abstract-gradient-training",
    author="Philip Sosnin",
    author_email="philipsosnin@gmail.com",
    packages=find_packages(),
    python_requires=">=3.10",
    install_requires=[
        "torch >= 2.4.1",
        "torchvision >= 0.19",
        "pydantic >= 2.9",
        "gurobipy >= 10.0",
        "numpy >= 1.9",
        "scipy >= 1.14",
        "pytest >= 8.3",
    ],
    extras_require={
        "examples": [
            "opacus >= 1.5",
            "seaborn >= 0.13",
            "matplotlib >= 3.9",
            "medmnist >= 3.0",
            "uci_datasets @ git+https://github.com/treforevans/uci_datasets.git",
            "matplotlib-label-lines >= 0.7",
        ],
    },
    platforms=["any"],
)