Distiller/setup.py at master · Cli212/Distiller · 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
42
43
44
45
46
47
48
49
50
51
52
from setuptools import setup, find_packages

setup(
    name = "distiller",
    version = "1.0",
    keywords = ("distiller", "knowledge distillation"),
    description = "eds sdk",
    long_description = "eds sdk for python",
    license = "MIT Licence",

    url = "http://test.com",
    author = "test",
    author_email = "test@gmail.com",

    packages=["Distiller"],
    package_dir={'': 'src'},
    include_package_data = True,
    platforms = "any",
    install_requires = [
        "scipy >= 1.0.0",
        "numpy>=1.17",
        "requests>=2.22.0",
        "Cython",
        "fsspec",
        "wheel",
        "dill",
        "pandas",
        "tqdm>=4.27",
        "xxhash",
        "multiprocess",
        "packaging",
        "regex!=2019.12.17",
        "tensorboard",
        "sacremoses",
        "pyemd",
        "sklearn",
        "fairseq",
        'sentencepiece',
        "fastBPE",
        "boto3",
        "nlpaug==1.1.3",
        "transformers==4.3.2",
        "datasets"
    ],

    scripts = [],
    entry_points = {
        'console_scripts': [
            'test = test.help:main'
        ]
    }
)