diff options
author | Arne Juul <arnej@yahooinc.com> | 2023-02-26 15:55:39 +0000 |
---|---|---|
committer | Arne Juul <arnej@yahooinc.com> | 2023-02-26 15:55:39 +0000 |
commit | 5716f46a8f4d635e8e375bef223a44ddd92077e1 (patch) | |
tree | de19b1d60ca3c56ce986ec2c2a5bfe54786ce21b /model-evaluation/src | |
parent | c16825629640c818d149061f21796de43c4f4455 (diff) |
replace assert -> check and throw
Diffstat (limited to 'model-evaluation/src')
-rw-r--r-- | model-evaluation/src/main/java/ai/vespa/models/evaluation/OnnxModel.java | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/model-evaluation/src/main/java/ai/vespa/models/evaluation/OnnxModel.java b/model-evaluation/src/main/java/ai/vespa/models/evaluation/OnnxModel.java index b86cf60318a..73c5eb36539 100644 --- a/model-evaluation/src/main/java/ai/vespa/models/evaluation/OnnxModel.java +++ b/model-evaluation/src/main/java/ai/vespa/models/evaluation/OnnxModel.java @@ -48,11 +48,13 @@ class OnnxModel implements AutoCloseable { final List<OutputSpec> outputSpecs = new ArrayList<>(); void addInputMapping(String onnxName, String source) { - assert(referencedEvaluator == null); + if (referencedEvaluator != null) + throw new IllegalStateException("input mapping must be added before load()"); inputSpecs.add(new InputSpec(onnxName, source)); } void addOutputMapping(String onnxName, String outputAs) { - assert(referencedEvaluator == null); + if (referencedEvaluator != null) + throw new IllegalStateException("output mapping must be added before load()"); outputSpecs.add(new OutputSpec(onnxName, outputAs)); } |