summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorHarald Musum <musum@yahooinc.com>2022-02-14 16:04:58 +0100
committerHarald Musum <musum@yahooinc.com>2022-02-14 16:04:58 +0100
commit34f8c11c57f7f2899f8c163e37c36e418261ca6d (patch)
treeed091833083f4fb1bb29c18dbad565d091ea9b94
parent2cc844d5050017b389492274e0b373610bd061ea (diff)
Skip tests if onnxruntime unavailable, part 3
-rw-r--r--application/src/test/java/com/yahoo/application/container/ContainerModelEvaluationTest.java3
1 files changed, 3 insertions, 0 deletions
diff --git a/application/src/test/java/com/yahoo/application/container/ContainerModelEvaluationTest.java b/application/src/test/java/com/yahoo/application/container/ContainerModelEvaluationTest.java
index 193fec3f7bb..f76008b4e02 100644
--- a/application/src/test/java/com/yahoo/application/container/ContainerModelEvaluationTest.java
+++ b/application/src/test/java/com/yahoo/application/container/ContainerModelEvaluationTest.java
@@ -1,6 +1,7 @@
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.application.container;
+import ai.vespa.modelintegration.evaluator.OnnxEvaluator;
import com.yahoo.application.Application;
import com.yahoo.application.Networking;
import com.yahoo.application.container.handler.Request;
@@ -17,6 +18,7 @@ import java.nio.charset.CharacterCodingException;
import java.nio.charset.StandardCharsets;
import static org.junit.Assert.assertEquals;
+import static org.junit.Assume.assumeTrue;
/**
* Verify that we can create a JDisc (and hence Application) instance capable of doing model evaluation
@@ -37,6 +39,7 @@ public class ContainerModelEvaluationTest {
@Test
public void testCreateApplicationInstanceWithModelEvaluation() {
+ assumeTrue(OnnxEvaluator.isRuntimeAvailable());
try (Application application =
Application.fromApplicationPackage(new File("src/test/app-packages/model-evaluation"),
Networking.disable)) {