From 3674b9c65f77638eaf46ec7241b5c8da6b1ada78 Mon Sep 17 00:00:00 2001
From: Varun Agrawal <varagrawal@gmail.com>
Date: Mon, 31 Aug 2020 09:51:14 -0400
Subject: [PATCH] Add data files to python package

---
 python/setup.py.in | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/python/setup.py.in b/python/setup.py.in
index a7acdc4ba..1ffe978f3 100644
--- a/python/setup.py.in
+++ b/python/setup.py.in
@@ -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,