aboutsummaryrefslogtreecommitdiffstats
path: root/model-integration/src/test/java/ai/vespa/embedding/BertBaseEmbedderTest.java
diff options
context:
space:
mode:
authorBjørn Christian Seime <bjorncs@yahooinc.com>2023-08-04 13:33:13 +0200
committerBjørn Christian Seime <bjorncs@yahooinc.com>2023-08-04 13:52:27 +0200
commit3ddce3c158941cb8eabdb391d207fe004095c434 (patch)
treedb8121fe016fa7bb5a0ad345af9f1bd58fbbfb6d /model-integration/src/test/java/ai/vespa/embedding/BertBaseEmbedderTest.java
parent0f46015e498ecb622473cd3e2403283c99f9f5d5 (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.java3
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);
}
}