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(),