From e82162d7f8d4ddd94a601eaf2d046420e6ec479e Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Mon, 15 Aug 2022 15:19:00 +0800 Subject: [PATCH] [backport] Fix Python package source install. (#8036) (#8171) * Copy gputreeshap. --- python-package/setup.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python-package/setup.py b/python-package/setup.py index 00f6ee775..a22b3eaf2 100644 --- a/python-package/setup.py +++ b/python-package/setup.py @@ -61,6 +61,7 @@ def copy_tree(src_dir: str, target_dir: str) -> None: src = os.path.join(src_dir, 'src') inc = os.path.join(src_dir, 'include') dmlc_core = os.path.join(src_dir, 'dmlc-core') + gputreeshap = os.path.join(src_dir, "gputreeshap") rabit = os.path.join(src_dir, 'rabit') cmake = os.path.join(src_dir, 'cmake') plugin = os.path.join(src_dir, 'plugin') @@ -68,6 +69,7 @@ def copy_tree(src_dir: str, target_dir: str) -> None: clean_copy_tree(src, os.path.join(target_dir, 'src')) clean_copy_tree(inc, os.path.join(target_dir, 'include')) clean_copy_tree(dmlc_core, os.path.join(target_dir, 'dmlc-core')) + clean_copy_tree(gputreeshap, os.path.join(target_dir, "gputreeshap")) clean_copy_tree(rabit, os.path.join(target_dir, 'rabit')) clean_copy_tree(cmake, os.path.join(target_dir, 'cmake')) clean_copy_tree(plugin, os.path.join(target_dir, 'plugin'))