diff options
author | Bjørn Christian Seime <bjorncs@verizonmedia.com> | 2022-02-14 16:32:14 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-02-14 16:32:14 +0100 |
commit | 7a863e7165c02acfb7d55b3e2fe5a1f645a312ae (patch) | |
tree | c6b75c0e390ff805d06effe7b56af002f6fdfb87 | |
parent | b4ec64e8fdb6803123a7b825379c0da7ff49a972 (diff) | |
parent | 34f8c11c57f7f2899f8c163e37c36e418261ca6d (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.java | 3 |
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)) { |