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(); }