diff options
author | Jon Bratseth <bratseth@gmail.com> | 2021-09-30 14:21:25 +0200 |
---|---|---|
committer | Jon Bratseth <bratseth@gmail.com> | 2021-09-30 14:21:25 +0200 |
commit | 380b9fa780ead9bcce0e824f7b6ee305e37dec43 (patch) | |
tree | 1a1d9a79910fdfc976643f02f1735d939ab689bf /linguistics-components/src/test/java/com/yahoo/language/sentencepiece | |
parent | 5007f6edcb5a2a461e859a10198f02171eab5516 (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.java | 4 |
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) { |