clean dir if archive failed

pytorch · Aug 26, 2023 · 4ee03ef · 4ee03ef
1 parent 1e1013b
commit 4ee03ef
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/frontend/archive/src/main/java/org/pytorch/serve/archive/model/ModelArchive.java b/frontend/archive/src/main/java/org/pytorch/serve/archive/model/ModelArchive.java
@@ -132,8 +132,12 @@ private static ModelArchive load(String url, File dir, boolean extracted)
             failed = false;
             return new ModelArchive(manifest, url, dir, extracted);
         } finally {
-            if (extracted && failed) {
-                FileUtils.deleteQuietly(dir);
+            if (failed) {
+                if (Files.isSymbolicLink(dir.toPath())) {
+                    FileUtils.deleteQuietly(dir.getParentFile());
+                } else {
+                    FileUtils.deleteQuietly(dir);
+                }
             }
         }
     }