diff options
author | Bjørn Christian Seime <bjorncs@yahooinc.com> | 2023-08-04 13:33:13 +0200 |
---|---|---|
committer | Bjørn Christian Seime <bjorncs@yahooinc.com> | 2023-08-04 13:52:27 +0200 |
commit | 3ddce3c158941cb8eabdb391d207fe004095c434 (patch) | |
tree | db8121fe016fa7bb5a0ad345af9f1bd58fbbfb6d /model-integration/src/test/java/ai/vespa/embedding/BertBaseEmbedderTest.java | |
parent | 0f46015e498ecb622473cd3e2403283c99f9f5d5 (diff) |
Add generic metrics for embedders
Diffstat (limited to 'model-integration/src/test/java/ai/vespa/embedding/BertBaseEmbedderTest.java')
-rw-r--r-- | model-integration/src/test/java/ai/vespa/embedding/BertBaseEmbedderTest.java | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/model-integration/src/test/java/ai/vespa/embedding/BertBaseEmbedderTest.java b/model-integration/src/test/java/ai/vespa/embedding/BertBaseEmbedderTest.java index 329b87cacd1..a0964eb5812 100644 --- a/model-integration/src/test/java/ai/vespa/embedding/BertBaseEmbedderTest.java +++ b/model-integration/src/test/java/ai/vespa/embedding/BertBaseEmbedderTest.java @@ -3,6 +3,7 @@ package ai.vespa.embedding; import ai.vespa.modelintegration.evaluator.OnnxRuntime; import com.yahoo.config.ModelReference; import com.yahoo.embedding.BertBaseEmbedderConfig; +import com.yahoo.language.process.Embedder; import com.yahoo.tensor.Tensor; import com.yahoo.tensor.TensorType; import org.junit.Test; @@ -69,7 +70,7 @@ public class BertBaseEmbedderTest { } private static BertBaseEmbedder newBertBaseEmbedder(BertBaseEmbedderConfig cfg) { - return new BertBaseEmbedder(new OnnxRuntime(), cfg); + return new BertBaseEmbedder(new OnnxRuntime(), Embedder.Runtime.testInstance(), cfg); } } |