12345qiupeng
/
gtsam
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add data files to python package

release/4.3a0
Varun Agrawal 2020-08-31 09:51:14 -04:00
parent a7f8ba7ac4
commit 3674b9c65f
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
python/setup.py.in
View File

@ -1,3 +1,4 @@
import glob
import os import os
import sys import sys
@ -8,6 +9,11 @@ except ImportError:
packages = find_packages(where=".") packages = find_packages(where=".")
print("PACKAGES: ", packages) print("PACKAGES: ", packages)
data_path = '${GTSAM_SOURCE_DIR}/examples/Data/'
data_files_and_directories = glob.glob(data_path + '**', recursive=True)
data_files = [x for x in data_files_and_directories if not os.path.isdir(x)]
package_data = { package_data = {
'': [ '': [
'./*.so', './*.so',
@ -44,6 +50,7 @@ setup(
], ],
packages=packages, packages=packages,
package_data=package_data, package_data=package_data,
data_files=[('${GTSAM_PYTHON_DATASET_DIR}', data_files),],
test_suite="gtsam.tests", test_suite="gtsam.tests",
install_requires=["numpy"], install_requires=["numpy"],
zip_safe=False, zip_safe=False,