path limit?

setup.py

@@ -1,10 +1,21 @@
 # !/usr/bin/env python
 
+import os
 from setuptools import setup, find_packages
 
+def package_files(directory):
+    paths = []
+    for (path, directories, filenames) in os.walk(directory):
+        for filename in filenames:
+            paths.append(os.path.join('..', path, filename))
+    return paths
+
+extra_files = package_files('WrightTools/datasets')
+
 setup(
     name='WrightTools',
     packages=find_packages(),
+    package_data={'': extra_files},
     install_requires=['h5py', 'matplotlib', 'numpy', 'python-dateutil', 'pytz', 'scipy'],
     version='2.13.4',
     description='Tools for loading, processing, and plotting multidimensional spectroscopy data.',