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(
long_description_content_type = 'text/markdown',
license = 'MIT',
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 = {
'nni_annotation': '../../tools/nni_annotation',
'nni_cmd': '../../tools/nni_cmd',
'nni_trial_tool': '../../tools/nni_trial_tool',
'nni_gpu_tool': '../../tools/nni_gpu_tool',
'nni': '../../src/sdk/pynni/nni'
},
python_requires = '>=3.5',
......
......@@ -48,7 +48,8 @@ setup(
'nni': 'src/sdk/pynni/nni',
'nni_annotation': 'tools/nni_annotation',
'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']},
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