aboutsummaryrefslogtreecommitdiffstats
path: root/linguistics-components/src/test/java/com/yahoo/language/sentencepiece
diff options
context:
space:
mode:
authorJon Bratseth <bratseth@gmail.com>2021-09-30 14:21:25 +0200
committerJon Bratseth <bratseth@gmail.com>2021-09-30 14:21:25 +0200
commit380b9fa780ead9bcce0e824f7b6ee305e37dec43 (patch)
tree1a1d9a79910fdfc976643f02f1735d939ab689bf /linguistics-components/src/test/java/com/yahoo/language/sentencepiece
parent5007f6edcb5a2a461e859a10198f02171eab5516 (diff)
Update linguisticvs-components
Diffstat (limited to 'linguistics-components/src/test/java/com/yahoo/language/sentencepiece')
-rw-r--r--linguistics-components/src/test/java/com/yahoo/language/sentencepiece/SentencePieceTester.java4
1 files changed, 2 insertions, 2 deletions
diff --git a/linguistics-components/src/test/java/com/yahoo/language/sentencepiece/SentencePieceTester.java b/linguistics-components/src/test/java/com/yahoo/language/sentencepiece/SentencePieceTester.java
index c4cb13a3d23..c6aa8fdd370 100644
--- a/linguistics-components/src/test/java/com/yahoo/language/sentencepiece/SentencePieceTester.java
+++ b/linguistics-components/src/test/java/com/yahoo/language/sentencepiece/SentencePieceTester.java
@@ -29,13 +29,13 @@ class SentencePieceTester {
}
public void assertEmbedded(String input, Integer... expectedCodes) {
- assertArrayEquals(expectedCodes, embedder.embed(input, Language.UNKNOWN).toArray());
+ assertArrayEquals(expectedCodes, embedder.embed(input, Language.UNKNOWN, null).toArray());
}
public void assertEmbedded(String input, String tensorType, String tensor) {
TensorType type = TensorType.fromSpec(tensorType);
Tensor expected = Tensor.from(type, tensor);
- assertEquals(expected, embedder.embed(input, Language.UNKNOWN, type));
+ assertEquals(expected, embedder.embed(input, Language.UNKNOWN, null, type));
}
public void assertSegmented(String input, String... expectedSegments) {