Skip to content

Commit

Permalink
include demo data and scripts in PyPI packaging
Browse files Browse the repository at this point in the history
  • Loading branch information
themattinthehatt committed Apr 1, 2024
1 parent 94b7bb3 commit 3d0363a
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand All @@ -103,4 +110,6 @@ def get_cuda_version():
author_email="[email protected]",
url="https://github.com/danbider/lightning-pose",
keywords=["machine learning", "deep learning", "computer_vision"],
package_data={"lightning_pose": data_files},
include_package_data=True,
)

0 comments on commit 3d0363a

Please sign in to comment.