Commit 92521826 authored by Yuxin Wu's avatar Yuxin Wu

use package data to include ops

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