summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorBjørn Christian Seime <bjorncs@verizonmedia.com>2022-02-14 16:32:14 +0100
committerGitHub <noreply@github.com>2022-02-14 16:32:14 +0100
commit7a863e7165c02acfb7d55b3e2fe5a1f645a312ae (patch)
treec6b75c0e390ff805d06effe7b56af002f6fdfb87
parentb4ec64e8fdb6803123a7b825379c0da7ff49a972 (diff)
parent34f8c11c57f7f2899f8c163e37c36e418261ca6d (diff)
Merge pull request #21178 from vespa-engine/hmusum/skip-onnx-tests-if-onnxruntime-unavailable-3
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)) {