From e889aa3db9d7df642db52d81349150ff1af15afe Mon Sep 17 00:00:00 2001
From: Konstantin Gulin <66528950+KSGulin@users.noreply.github.com>
Date: Wed, 22 Jun 2022 15:00:48 +0100
Subject: [PATCH] Include YOLOv5 data files in sparseml wheel (#904) (#906)

---
 MANIFEST.in | 3 ++-
 setup.py    | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/MANIFEST.in b/MANIFEST.in
index cc0d11642bc..80876be44a6 100644
--- a/MANIFEST.in
+++ b/MANIFEST.in
@@ -1 +1,2 @@
-include LICENSE
\ No newline at end of file
+include LICENSE
+recursive-include src/sparseml/yolov5 *.yaml *.sh
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 9d6a59ba3b9..4ee1ef15eac 100644
--- 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(),