diff --git a/setup.py b/setup.py index 435651be4..a09d2714d 100644 --- a/setup.py +++ b/setup.py @@ -75,6 +75,13 @@ def run(self): 'visualdl.server.onnx', ] +datas = [] +data_root = os.path.join(TOP_DIR, 'visualdl/server/dist') +for root, dirs, files in os.walk(data_root): + for filename in files: + path = os.path.join(root, filename)[len(data_root)+1:] + datas.append(path) + setup( name="visualdl", version=VERSION_NUMBER, @@ -84,7 +91,7 @@ def run(self): keywords="visualization deeplearning", long_description=read('README.md'), install_requires=install_requires, - package_data={'visualdl.server': ['dist/*', 'dist/fonts/*'], + package_data={'visualdl.server': datas, 'visualdl':['core.so'], 'visualdl.python':['core.so']}, packages=packages,