Fix runtime error on osx (apache#1449)

sergei-mironov · Aug 8, 2018 · 578a528 · 578a528
1 parent 07e3888
commit 578a528
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/python/setup.py b/python/setup.py
@@ -104,10 +104,11 @@ def is_pure(self):
 
 # For bdist_wheel only
 if wheel_include_libs:
-    for path in LIB_LIST:
-        shutil.copy(path, os.path.join(CURRENT_DIR, 'tvm'))
-        _, libname = os.path.split(path)
-        fo.write("include tvm/%s\n" % libname)
+    with open("MANIFEST.in", "w") as fo:
+        for path in LIB_LIST:
+            shutil.copy(path, os.path.join(CURRENT_DIR, 'tvm'))
+            _, libname = os.path.split(path)
+            fo.write("include tvm/%s\n" % libname)
     setup_kwargs = {
         "include_package_data": True
     }
@@ -118,7 +119,7 @@ def is_pure(self):
         LIB_LIST[i] = os.path.relpath(path, curr_path)
     setup_kwargs = {
         "include_package_data": True,
-        "package_data": {'tvm': LIB_LIST}
+        "data_files": [('tvm', LIB_LIST)]
     }
 
 setup(name='tvm',