diff options
Diffstat (limited to 'config-model-api/src/main/java/com/yahoo/config/model/api/OnnxModelCost.java')
-rw-r--r-- | config-model-api/src/main/java/com/yahoo/config/model/api/OnnxModelCost.java | 14 |
1 files changed, 2 insertions, 12 deletions
diff --git a/config-model-api/src/main/java/com/yahoo/config/model/api/OnnxModelCost.java b/config-model-api/src/main/java/com/yahoo/config/model/api/OnnxModelCost.java index 4764d501865..1efd98184cc 100644 --- a/config-model-api/src/main/java/com/yahoo/config/model/api/OnnxModelCost.java +++ b/config-model-api/src/main/java/com/yahoo/config/model/api/OnnxModelCost.java @@ -8,10 +8,10 @@ import com.yahoo.config.provision.ClusterSpec; import java.net.URI; import java.util.Map; -import java.util.Optional; /** * @author bjorncs + * @author hmusum */ public interface OnnxModelCost { @@ -31,17 +31,7 @@ public interface OnnxModelCost { void store(); } - record ModelInfo(String modelId, long estimatedCost, long hash, Optional<OnnxModelOptions> onnxModelOptions) { - - public ModelInfo(String modelId, long estimatedCost, long hash, OnnxModelOptions onnxModelOptions) { - this(modelId, estimatedCost, hash, Optional.of(onnxModelOptions)); - } - - public OnnxModelOptions options() { - return onnxModelOptions.orElseThrow(() -> new IllegalStateException("No onnxModelOptions exist")); - } - - } + record ModelInfo(String modelId, long estimatedCost, long hash, OnnxModelOptions onnxModelOptions) {} static OnnxModelCost disabled() { return new DisabledOnnxModelCost(); } |