diff options
author | Harald Musum <musum@yahooinc.com> | 2022-02-14 16:04:58 +0100 |
---|---|---|
committer | Harald Musum <musum@yahooinc.com> | 2022-02-14 16:04:58 +0100 |
commit | 34f8c11c57f7f2899f8c163e37c36e418261ca6d (patch) | |
tree | ed091833083f4fb1bb29c18dbad565d091ea9b94 | |
parent | 2cc844d5050017b389492274e0b373610bd061ea (diff) |
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)) { |