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

View File

@ -1,3 +1,4 @@
import glob
import os
import sys
@ -8,6 +9,11 @@ except ImportError:
packages = find_packages(where=".")
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 = {
'': [
'./*.so',
@ -44,6 +50,7 @@ setup(
],
packages=packages,
package_data=package_data,
data_files=[('${GTSAM_PYTHON_DATASET_DIR}', data_files),],
test_suite="gtsam.tests",
install_requires=["numpy"],
zip_safe=False,