diff options
author | Lester Solbakken <lester.solbakken@gmail.com> | 2024-04-16 13:46:49 +0200 |
---|---|---|
committer | Lester Solbakken <lester.solbakken@gmail.com> | 2024-04-16 13:46:49 +0200 |
commit | 44e36267bf12cdaba80ae453c94429ebefc34cde (patch) | |
tree | 4aee25b66d4df6deee7d6c8514d641a7c805afeb | |
parent | 8d69cbfb97cc4047f1e952ddbef7c52a74d5a2ef (diff) |
Disable local LLM unit tests
-rw-r--r-- | model-integration/src/test/java/ai/vespa/llm/clients/LocalLLMTest.java | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/model-integration/src/test/java/ai/vespa/llm/clients/LocalLLMTest.java b/model-integration/src/test/java/ai/vespa/llm/clients/LocalLLMTest.java index e85e397b7ff..a3b260f3fb5 100644 --- a/model-integration/src/test/java/ai/vespa/llm/clients/LocalLLMTest.java +++ b/model-integration/src/test/java/ai/vespa/llm/clients/LocalLLMTest.java @@ -6,6 +6,7 @@ import ai.vespa.llm.completion.Completion; import ai.vespa.llm.completion.Prompt; import ai.vespa.llm.completion.StringPrompt; import com.yahoo.config.ModelReference; +import org.junit.jupiter.api.Disabled; import org.junit.jupiter.api.Test; import java.util.ArrayList; @@ -32,6 +33,7 @@ public class LocalLLMTest { private static Prompt prompt = StringPrompt.from("A random prompt"); @Test + @Disabled public void testGeneration() { var config = new LlmLocalClientConfig.Builder() .parallelRequests(1) @@ -48,6 +50,7 @@ public class LocalLLMTest { } @Test + @Disabled public void testAsyncGeneration() { var sb = new StringBuilder(); var tokenCount = new AtomicInteger(0); @@ -71,10 +74,11 @@ public class LocalLLMTest { llm.deconstruct(); } assertTrue(tokenCount.get() > 0); -// System.out.println(sb); + System.out.println(sb); } @Test + @Disabled public void testParallelGeneration() { var prompts = testPrompts(); var promptsToUse = prompts.size(); @@ -113,6 +117,7 @@ public class LocalLLMTest { } @Test + @Disabled public void testRejection() { var prompts = testPrompts(); var promptsToUse = prompts.size(); |