Commit 00b03d2e authored by Zejun Lin's avatar Zejun Lin Committed by chicm-ms
Browse files

Add package_data in setup.py (#795)

add package requirement data in setup.py
parent bf72b298
...@@ -56,6 +56,7 @@ setuptools.setup( ...@@ -56,6 +56,7 @@ setuptools.setup(
'nni_gpu_tool': '../../tools/nni_gpu_tool', 'nni_gpu_tool': '../../tools/nni_gpu_tool',
'nni': '../../src/sdk/pynni/nni' 'nni': '../../src/sdk/pynni/nni'
}, },
package_data = {'nni': ['**/requirements.txt']},
python_requires = '>=3.5', python_requires = '>=3.5',
install_requires = [ install_requires = [
'schema', 'schema',
...@@ -81,4 +82,4 @@ setuptools.setup( ...@@ -81,4 +82,4 @@ setuptools.setup(
'nnictl = nni_cmd.nnictl:parse_args' 'nnictl = nni_cmd.nnictl:parse_args'
] ]
} }
) )
\ No newline at end of file
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