Fix setup.py backwards compatibiity (#586)

* Fix setup.py backwards compatibiity * Remove version check
2022-02-22 10:54:05 -05:00 · 2022-02-22 10:54:05 -05:00 · 7023ee53e8
parent ea371a9bf2
commit 7023ee53e8
1 changed files with 5 additions and 2 deletions
--- a/setup.py
+++ b/setup.py
@ -83,10 +83,13 @@ class CMakeBuild(build_py):
        python_package_dir = os.path.join(cmake_build_dir,
                                          "tools", "torch-mlir", "python_packages",
                                          "torch_mlir")
+
+        if os.path.exists(target_dir):
+            shutil.rmtree(target_dir, ignore_errors=False, onerror=None)
+
        shutil.copytree(python_package_dir,
                        target_dir,
-                        symlinks=False,
-                        dirs_exist_ok=True)
+                        symlinks=False)


 class CMakeExtension(Extension):