Fix M5 proofpoint

Fix the issue where the pretrained_models and datasets directory was not being copied with pip install not using -e.
groq · Mar 5, 2024 · c9ea645 · c9ea645
1 parent 8082a8c
commit c9ea645
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/demo_helpers/MANIFEST.in b/demo_helpers/MANIFEST.in
@@ -0,0 +1,3 @@
+include demo_helpers/datasets/README.md
+include demo_helpers/pretrained_models/m5.pt
+include demo_helpers/pretrained_models/pointnet.pth
diff --git a/demo_helpers/setup.py b/demo_helpers/setup.py
@@ -10,6 +10,7 @@
     packages=find_packages(
         exclude=["*.__pycache__.*"],
     ),
+    include_package_data=True,
     install_requires=[
         "charset-normalizer==3.3.2",
         "transformers>=4.20.0",