From 3d0363ae0609f46ddccc547e16d4fa3e9e10152a Mon Sep 17 00:00:00 2001 From: themattinthehatt Date: Mon, 1 Apr 2024 17:10:02 -0400 Subject: [PATCH] include demo data and scripts in PyPI packaging --- setup.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/setup.py b/setup.py index a96449a8..67666801 100644 --- a/setup.py +++ b/setup.py @@ -89,6 +89,13 @@ def get_cuda_version(): }, } +# collect all data and script files +data_files = [] +for root, dirs, files in os.walk("lightning_pose/data"): + data_files.extend([join(root, f) for f in files]) +for root, dirs, files in os.walk("lightning_pose/scripts"): + data_files.extend([join(root, f) for f in files]) + setup( name="lightning-pose", @@ -103,4 +110,6 @@ def get_cuda_version(): author_email="danbider@gmail.com", url="https://github.com/danbider/lightning-pose", keywords=["machine learning", "deep learning", "computer_vision"], + package_data={"lightning_pose": data_files}, + include_package_data=True, )