Skip to content

Commit c8bc1b2

Browse files
Merge pull request #228 from bernardladenthin/claude/confident-babbage-ggbty5
Skip reranking tests when model file is unavailable
2 parents f8a62dd + b4d16b9 commit c8bc1b2

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

src/test/java/net/ladenthin/llama/RerankingModelTest.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,12 +7,14 @@
77

88
import static org.junit.jupiter.api.Assertions.*;
99

10+
import java.io.File;
1011
import java.util.List;
1112
import java.util.Map;
1213
import net.ladenthin.llama.parameters.ModelParameters;
1314
import net.ladenthin.llama.value.LlamaOutput;
1415
import net.ladenthin.llama.value.Pair;
1516
import org.junit.jupiter.api.AfterAll;
17+
import org.junit.jupiter.api.Assumptions;
1618
import org.junit.jupiter.api.BeforeAll;
1719
import org.junit.jupiter.api.Test;
1820

@@ -30,6 +32,9 @@ public class RerankingModelTest {
3032

3133
@BeforeAll
3234
public static void setup() {
35+
Assumptions.assumeTrue(
36+
new File("models/jina-reranker-v1-tiny-en-Q4_0.gguf").exists(),
37+
"Reranking model not available, skipping tests");
3338
int gpuLayers = Integer.getInteger(TestConstants.PROP_TEST_NGL, TestConstants.DEFAULT_TEST_NGL);
3439
model = new LlamaModel(new ModelParameters()
3540
.setCtxSize(128)

0 commit comments

Comments
 (0)