summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorLester Solbakken <lester.solbakken@gmail.com>2024-04-16 13:46:49 +0200
committerLester Solbakken <lester.solbakken@gmail.com>2024-04-16 13:46:49 +0200
commit44e36267bf12cdaba80ae453c94429ebefc34cde (patch)
tree4aee25b66d4df6deee7d6c8514d641a7c805afeb
parent8d69cbfb97cc4047f1e952ddbef7c52a74d5a2ef (diff)
Disable local LLM unit tests
-rw-r--r--model-integration/src/test/java/ai/vespa/llm/clients/LocalLLMTest.java7
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();