From d5dda1ebf2d17173806ad89fe639ed55da676b8f Mon Sep 17 00:00:00 2001 From: Philipp Moritz Date: Thu, 2 Aug 2018 17:06:37 -0700 Subject: [PATCH] copy all files when installing pyarrow (#2547) --- python/setup.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/python/setup.py b/python/setup.py index 8421398ff..050b90dd7 100644 --- a/python/setup.py +++ b/python/setup.py @@ -77,12 +77,10 @@ class build_ext(_build_ext.build_ext): # We also need to install pyarrow along with Ray, so make sure that the # relevant non-Python pyarrow files get copied. - pyarrow_files = [ - os.path.join("ray/pyarrow_files/pyarrow", filename) - for filename in os.listdir("./ray/pyarrow_files/pyarrow") - if not os.path.isdir( - os.path.join("ray/pyarrow_files/pyarrow", filename)) - ] + pyarrow_files = [] + for (root, dirs, filenames) in os.walk("./ray/pyarrow_files/pyarrow"): + for name in filenames: + pyarrow_files.append(os.path.join(root, name)) files_to_include = ray_files + pyarrow_files