From 38721b233d8762e862773c8b5f2cf61ca4ea3769 Mon Sep 17 00:00:00 2001 From: Henning Baldersheim Date: Fri, 30 Jun 2023 10:02:11 +0200 Subject: No need to close an evaluater that has not been created(loaded) --- .../src/main/java/ai/vespa/models/evaluation/OnnxModel.java | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'model-evaluation/src') 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; + } + } } -- cgit v1.2.3