[api] Avoid exception if SecurityManager is applied

deepjavalibrary · Jun 20, 2023 · f7ab0f1 · f7ab0f1
1 parent 8f841e6
commit f7ab0f1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/api/src/main/java/ai/djl/util/Ec2Utils.java b/api/src/main/java/ai/djl/util/Ec2Utils.java
@@ -51,7 +51,12 @@ private Ec2Utils() {}
      * @return true if the current environment is SageMaker
      */
     public static boolean isSageMaker() {
-        return Files.exists(Paths.get(ENDPOINT_METADATA_FILE));
+        try {
+            return Files.exists(Paths.get(ENDPOINT_METADATA_FILE));
+        } catch (SecurityException e) {
+            logger.warn("Security manager doesn't allow access file");
+            return false;
+        }
     }
 
     /**