Include YOLOv5 data files in sparseml wheel (#904) (#906)

neuralmagic · Jun 22, 2022 · e889aa3 · e889aa3
1 parent 9902945
commit e889aa3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -1 +1,2 @@
-include LICENSE
+include LICENSE
+recursive-include src/sparseml/yolov5 *.yaml *.sh
diff --git a/setup.py b/setup.py
@@ -203,6 +203,7 @@ def _setup_long_description() -> Tuple[str, str]:
     ),
     license="Apache",
     url="https://github.com/neuralmagic/sparseml",
+    include_package_data=True,
     package_dir=_setup_package_dir(),
     packages=_setup_packages(),
     install_requires=_setup_install_requires(),