From 4ffb2bdb6b82a9acc32fa40cf0be1b18af74126b Mon Sep 17 00:00:00 2001 From: Chun-Wei Chen Date: Wed, 14 Sep 2022 15:06:45 -0700 Subject: [PATCH] Use find_namespace_packages to get needed subdirectories to silent lots of warnings (#4457) * use find_namespace_packages instead of find_packages Signed-off-by: Chun-Wei Chen * specifiy data Signed-off-by: Chun-Wei Chen * specify needed dirs Signed-off-by: Chun-Wei Chen * fix flake8 Signed-off-by: Chun-Wei Chen * remove unnecessary set Signed-off-by: Chun-Wei Chen * fix black Signed-off-by: Chun-Wei Chen Signed-off-by: Chun-Wei Chen --- setup.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 8410fefabde..954b9fda590 100644 --- a/setup.py +++ b/setup.py @@ -305,8 +305,21 @@ def run(self): # Packages ################################################################################ -# no need to do fancy stuff so far -packages = setuptools.find_packages() +# Add package directories here if you want to package them with the source +include_dirs = [ + "onnx.backend.test.cpp*", + "onnx.backend.test.data.*", + "onnx.common", + "onnx.defs.*", + "onnx.examples*", + "onnx.shape_inference", + "onnx.test.cpp", + "onnx.version_converter*", +] + +packages = setuptools.find_packages() + setuptools.find_namespace_packages( + include=include_dirs +) requirements_file = "requirements.txt" requirements_path = os.path.join(os.getcwd(), requirements_file)