"vscode:/vscode.git/clone" did not exist on "d1ab802ce6cc16b01ce947fe1c0b0f5360af9b16"
setup.py 1.41 KB
Newer Older
1
2
3
4
import setuptools
from os import walk, path

data_files = [('bin', ['node-linux-x64/bin/node'])]
Gems Guo's avatar
Gems Guo committed
5
for (dirpath, dirnames, filenames) in walk('./nni'):
6
7
8
9
10
11
12
    files = [path.normpath(path.join(dirpath, filename)) for filename in filenames]
    data_files.append((path.normpath(dirpath), files))

with open('../../README.md', 'r') as fh:
    long_description = fh.read()

setuptools.setup(
Gems Guo's avatar
Gems Guo committed
13
    name = 'nni',
Gems Guo's avatar
Gems Guo committed
14
    version = '0.3.2',
15
16
17
18
19
20
21
    author = 'Microsoft NNI team',
    author_email = 'nni@microsoft.com',
    description = 'Neural Network Intelligence package',
    long_description = long_description,
    long_description_content_type = 'text/markdown',
    license = 'MIT',
    url = 'https://github.com/Microsoft/nni',
Gems Guo's avatar
Gems Guo committed
22
23
24
25
26
27
    packages = setuptools.find_packages('../../tools'),
    package_dir = {
        'nni_annotation': '../../tools/nni_annotation',
        'nni_cmd': '../../tools/nni_cmd',
        'nni_trial_tool':'../../tools/nni_trial_tool'
    },
28
29
    python_requires = '>=3.5',
    install_requires = [
Gems Guo's avatar
Gems Guo committed
30
        'nni-sdk',
Gems Guo's avatar
Gems Guo committed
31
32
33
34
        'schema',
        'pyyaml',
        'psutil',
        'requests',
Gems Guo's avatar
Gems Guo committed
35
        'astor'
36
37
38
39
40
41
42
43
44
    ],
    classifiers = [
        'Programming Language :: Python :: 3',
        'License :: OSI Approved :: MIT License',
        'Operating System :: POSIX :: Linux'
    ],
    data_files = data_files,
    entry_points = {
        'console_scripts' : [
Gems Guo's avatar
Gems Guo committed
45
            'nnictl = nni_cmd.nnictl:parse_args'
46
47
48
        ]
    }
)