You need to sign in or sign up before continuing.
Commit 92521826 authored by Yuxin Wu's avatar Yuxin Wu

use package data to include ops

parent f877b6e9
include requirements.txt include requirements.txt
include tensorpack/user_ops/Makefile
include tensorpack/user_ops/*.cc
include tensorpack/user_ops/*.h
...@@ -27,10 +27,15 @@ setup( ...@@ -27,10 +27,15 @@ setup(
version=__version__, version=__version__,
description='Neural Network Toolbox on TensorFlow', description='Neural Network Toolbox on TensorFlow',
long_description=long_description, long_description=long_description,
install_requires=req, install_requires=req,
tests_require=['flake8', 'scikit-image'], tests_require=['flake8', 'scikit-image'],
extras_require={ extras_require={
'all': ['pillow', 'scipy', 'h5py', 'lmdb>=0.92', 'matplotlib', 'all': ['pillow', 'scipy', 'h5py', 'lmdb>=0.92', 'matplotlib',
'scikit-learn', "tornado; python_version < '3.0'"] 'scikit-learn', "tornado; python_version < '3.0'"]
}, },
include_package_data=True,
package_data={'tensorpack': ['user_ops/Makefile', 'user_ops/*.cc', 'user_ops/*.h']},
) )
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