chore: add config.yaml when setup

RapidAI · Dec 15, 2024 · b29e46d · b29e46d
1 parent 714b019
commit b29e46d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup_undistort.py b/setup_undistort.py
@@ -43,9 +43,9 @@ def read_txt(txt_path: Union[Path, str]) -> List[str]:
     url="https://github.com/Joker1212/RapidTableDetection",
     license="Apache-2.0",
     install_requires=read_txt("requirements.txt"),
-    include_package_data=False,
+    include_package_data=True,
     packages=[MODULE_NAME, f"{MODULE_NAME}.models", f"{MODULE_NAME}.utils"],
-    package_data={"": [".gitkeep"]},
+    package_data={"": [".gitkeep"], MODULE_NAME: ["config.yaml"]},
     keywords=["obj detection,ocr,table-recognition"],
     classifiers=[
         "Programming Language :: Python :: 3.8",