diff options
author | Henning Baldersheim <balder@yahoo-inc.com> | 2022-12-12 14:53:00 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-12-12 14:53:00 +0100 |
commit | 2c4dd1e7ce7755259a994250aa246ab09a9da278 (patch) | |
tree | 099e17fd31f647f51c3a27e0ab9fa9b46e923522 | |
parent | 43993948c888686c895e88264965173d27bb14d5 (diff) | |
parent | d6e2c1f4392a790480de48798a8c16a016bbc216 (diff) |
Merge pull request #25214 from vespa-engine/arnej/empty-string-as-model-would-fail
trying to make an OnnxEvaluator with empty path would fail
-rw-r--r-- | model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java b/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java index bdcceddb04f..125707c9aaa 100644 --- a/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java +++ b/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java @@ -37,6 +37,9 @@ public class OnnxEvaluator { environment = OrtEnvironment.getEnvironment(); session = environment.createSession(modelPath, options.getOptions()); } catch (OrtException e) { + if (e.getCode() == OrtException.OrtErrorCode.ORT_NO_SUCHFILE) { + throw new IllegalArgumentException("No such file: "+modelPath); + } throw new RuntimeException("ONNX Runtime exception", e); } } @@ -101,6 +104,11 @@ public class OnnxEvaluator { try { new OnnxEvaluator(modelPath); return true; + } catch (IllegalArgumentException e) { + if (e.getMessage().equals("No such file: ")) { + return true; + } + return false; } catch (UnsatisfiedLinkError | RuntimeException | NoClassDefFoundError e) { return false; } |