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-api/src/main/java/com/yahoo | |
parent | f10979f97f25a7d989dc26b958bd2321f258112e (diff) |
Add ModelInfo constructor with non-optional OnnxModelOptions
Diffstat (limited to 'config-model-api/src/main/java/com/yahoo')
-rw-r--r-- | config-model-api/src/main/java/com/yahoo/config/model/api/OnnxModelCost.java | 14 |
1 files changed, 13 insertions, 1 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 d70b751eba0..650ce1ac14c 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 @@ -18,8 +18,10 @@ public interface OnnxModelCost { interface Calculator { long aggregatedModelCostInBytes(); + // TODO: Unused, remove when 8.263.7 is oldest model in use void registerModel(ApplicationFile path); void registerModel(ApplicationFile path, OnnxModelOptions onnxModelOptions); + // TODO: Unused, remove when 8.263.7 is oldest model in use void registerModel(URI uri); void registerModel(URI uri, OnnxModelOptions onnxModelOptions); Map<String, ModelInfo> models(); @@ -28,7 +30,17 @@ public interface OnnxModelCost { void store(); } - record ModelInfo(String modelId, long estimatedCost, long hash, Optional<OnnxModelOptions> onnxModelOptions) {} + 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")); + } + + } static OnnxModelCost disabled() { return new DisabledOnnxModelCost(); } |