fltk/setup.py at main · bacox/fltk · 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
36
37
38
39
40
41
from setuptools import setup, find_packages
from fltk import __version__
setup(
    name="fltk",
    author="Bart Cox",
    author_email="b.a.cox@tudelft.nl",
    maintainer="Bart Cox",
    maintainer_email="b.a.cox@tudelft.nl",
    description="Federated Learning Toolkit",
    packages=find_packages(),
    version=__version__,
    entry_points={
        "console_scripts": [
            "fltk = fltk.__main__:main",
        ]
    },
    include_package_data=True,
    data_files=[('share/tudelft/fltk/experiments', ['experiments/experiment.yaml'])],
    install_requires=
    [
        'tqdm==4.49.0',
        'scikit-learn==0.23.2',
        'pandas==1.1.2',
        'numpy>=1.20.0',
        'torch==1.10.1',
        'torchvision==0.11.2',
        'scipy==1.4.1',
        'h5py==2.10.0',
        'requests',
        'pyyaml',
        'torchsummary',
        'dataclass-csv',
        'tensorboard',
        'seaborn',
        'matplotlib',
        'google-api-python-client',
        'google-auth-httplib2',
        'google-auth-oauthlib >= 0.4.1',
        'spacy'
    ]
)