Unverified Commit 85d936d1 authored by SparkSnail's avatar SparkSnail Committed by GitHub
Browse files

Fix pypi package (#727)

* add nni_gpu_tool in setup.py
* add exclude=['test'] in setup.py
parent 6f3a90eb
...@@ -48,11 +48,12 @@ setuptools.setup( ...@@ -48,11 +48,12 @@ setuptools.setup(
long_description_content_type = 'text/markdown', long_description_content_type = 'text/markdown',
license = 'MIT', license = 'MIT',
url = 'https://github.com/Microsoft/nni', url = 'https://github.com/Microsoft/nni',
packages = setuptools.find_packages('../../tools') + setuptools.find_packages('../../src/sdk/pynni') packages = setuptools.find_packages('../../tools') + setuptools.find_packages('../../src/sdk/pynni', exclude=['tests']),
package_dir = { package_dir = {
'nni_annotation': '../../tools/nni_annotation', 'nni_annotation': '../../tools/nni_annotation',
'nni_cmd': '../../tools/nni_cmd', 'nni_cmd': '../../tools/nni_cmd',
'nni_trial_tool': '../../tools/nni_trial_tool', 'nni_trial_tool': '../../tools/nni_trial_tool',
'nni_gpu_tool': '../../tools/nni_gpu_tool',
'nni': '../../src/sdk/pynni/nni' 'nni': '../../src/sdk/pynni/nni'
}, },
python_requires = '>=3.5', python_requires = '>=3.5',
......
...@@ -48,7 +48,8 @@ setup( ...@@ -48,7 +48,8 @@ setup(
'nni': 'src/sdk/pynni/nni', 'nni': 'src/sdk/pynni/nni',
'nni_annotation': 'tools/nni_annotation', 'nni_annotation': 'tools/nni_annotation',
'nni_cmd': 'tools/nni_cmd', 'nni_cmd': 'tools/nni_cmd',
'nni_trial_tool':'tools/nni_trial_tool' 'nni_trial_tool':'tools/nni_trial_tool',
'nni_gpu_tool':'tools/nni_gpu_tool'
}, },
package_data = {'nni': ['**/requirements.txt']}, package_data = {'nni': ['**/requirements.txt']},
python_requires = '>=3.5', python_requires = '>=3.5',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment