copy all files when installing pyarrow (ray-project#2547)

roshansridhar · Aug 3, 2018 · d5dda1e · d5dda1e
1 parent 5e59cc6
commit d5dda1e
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/python/setup.py b/python/setup.py
@@ -77,12 +77,10 @@ def run(self):
 
         # 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