diff options
author | Harald Musum <musum@yahooinc.com> | 2024-01-02 11:18:12 +0100 |
---|---|---|
committer | Harald Musum <musum@yahooinc.com> | 2024-01-02 11:18:12 +0100 |
commit | 9a4fc8eb55df849ea823909b34338506e7a1b44f (patch) | |
tree | ab3780487dcc7e9625b13debb02b44718bd1a151 /config-model/src/test | |
parent | f10979f97f25a7d989dc26b958bd2321f258112e (diff) |
Add ModelInfo constructor with non-optional OnnxModelOptions
Diffstat (limited to 'config-model/src/test')
-rw-r--r-- | config-model/src/test/java/com/yahoo/vespa/model/application/validation/change/RestartOnDeployForOnnxModelChangesValidatorTest.java | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/config-model/src/test/java/com/yahoo/vespa/model/application/validation/change/RestartOnDeployForOnnxModelChangesValidatorTest.java b/config-model/src/test/java/com/yahoo/vespa/model/application/validation/change/RestartOnDeployForOnnxModelChangesValidatorTest.java index adcf58785fa..d0f5f01d61e 100644 --- a/config-model/src/test/java/com/yahoo/vespa/model/application/validation/change/RestartOnDeployForOnnxModelChangesValidatorTest.java +++ b/config-model/src/test/java/com/yahoo/vespa/model/application/validation/change/RestartOnDeployForOnnxModelChangesValidatorTest.java @@ -15,7 +15,6 @@ import java.net.URI; import java.util.HashMap; import java.util.List; import java.util.Map; -import java.util.Optional; import static org.junit.jupiter.api.Assertions.assertEquals; import static org.junit.jupiter.api.Assertions.assertTrue; @@ -99,7 +98,7 @@ public class RestartOnDeployForOnnxModelChangesValidatorTest { @Override public void registerModel(URI uri, OnnxModelOptions onnxModelOptions) { - models.put(uri.toString(), new OnnxModelCost.ModelInfo(uri.toString(), estimatedCost, hash, Optional.ofNullable(onnxModelOptions))); + models.put(uri.toString(), new OnnxModelCost.ModelInfo(uri.toString(), estimatedCost, hash, onnxModelOptions)); } @Override |