diff options
author | Henning Baldersheim <balder@yahoo-inc.com> | 2023-06-30 10:02:11 +0200 |
---|---|---|
committer | Henning Baldersheim <balder@yahoo-inc.com> | 2023-06-30 10:02:11 +0200 |
commit | 38721b233d8762e862773c8b5f2cf61ca4ea3769 (patch) | |
tree | aa11cbedc1818935b96d9bc954115ce149658b7c /model-evaluation | |
parent | 902931e48415b354c9790b41c28d40dfa6f076d5 (diff) |
No need to close an evaluater that has not been created(loaded)
Diffstat (limited to 'model-evaluation')
-rw-r--r-- | model-evaluation/src/main/java/ai/vespa/models/evaluation/OnnxModel.java | 9 |
1 files changed, 6 insertions, 3 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 ad27f9d2d15..f104bca20b9 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 @@ -4,8 +4,6 @@ package ai.vespa.models.evaluation; import ai.vespa.modelintegration.evaluator.OnnxEvaluator; import ai.vespa.modelintegration.evaluator.OnnxEvaluatorOptions; import ai.vespa.modelintegration.evaluator.OnnxRuntime; -import com.yahoo.searchlib.rankingexpression.Reference; -import com.yahoo.searchlib.rankingexpression.rule.CompositeNode; import com.yahoo.searchlib.rankingexpression.rule.ExpressionNode; import com.yahoo.tensor.Tensor; import com.yahoo.tensor.TensorType; @@ -207,5 +205,10 @@ class OnnxModel implements AutoCloseable { return evaluator; } - @Override public void close() { evaluator.close(); } + @Override public void close() { + if (evaluator != null) { + evaluator.close(); + evaluator = null; + } + } } |