Fix setup.py backwards compatibiity (llvm#586)

* Fix setup.py backwards compatibiity * Remove version check
Abhishek-TyRnT · Feb 22, 2022 · 7023ee5 · 7023ee5
1 parent ea371a9
commit 7023ee5
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -83,10 +83,13 @@ def run(self):
         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):