From 380b9fa780ead9bcce0e824f7b6ee305e37dec43 Mon Sep 17 00:00:00 2001 From: Jon Bratseth Date: Thu, 30 Sep 2021 14:21:25 +0200 Subject: Update linguisticvs-components --- .../java/com/yahoo/language/sentencepiece/SentencePieceTester.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'linguistics-components/src/test/java') 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) { -- cgit v1.2.3