add validation funcion

pytorch · Oct 6, 2023 · eae1ff8 · eae1ff8
1 parent 2c9f6c3
commit eae1ff8
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/frontend/server/src/main/java/org/pytorch/serve/wlm/ModelManager.java b/frontend/server/src/main/java/org/pytorch/serve/wlm/ModelManager.java
@@ -239,10 +239,7 @@ private void setupModelDependencies(Model model)
                             null);
 
             ProcessBuilder processBuilder = new ProcessBuilder(commandParts);
-            if (dependencyPath
-                    .toPath()
-                    .normalize()
-                    .startsWith(FileUtils.getTempDirectory().toPath().normalize())) {
+            if (isValidDependencyPath(dependencyPath)) {
                 processBuilder.directory(dependencyPath);
             } else {
                 throw new ModelException(
@@ -285,6 +282,16 @@ private void setupModelDependencies(Model model)
         }
     }
 
+    private boolean isValidDependencyPath(File dependencyPath) {
+        if (dependencyPath
+                .toPath()
+                .normalize()
+                .startsWith(FileUtils.getTempDirectory().toPath().normalize())) {
+            return true;
+        }
+        return false;
+    }
+
     private Model createModel(
             ModelArchive archive,
             int batchSize,