Remove 'required' attribute

1 files changed, 4 insertions, 12 deletions

diff --git a/config-model/src/main/java/com/yahoo/schema/OnnxModel.java b/config-model/src/main/java/com/yahoo/schema/OnnxModel.java
index ae6f1fd96e4..3d96849fa15 100644
--- a/config-model/src/main/java/com/yahoo/schema/OnnxModel.java
+++ b/config-model/src/main/java/com/yahoo/schema/OnnxModel.java
@@ -24,7 +24,7 @@ public class OnnxModel extends DistributableResource {
     private String  statelessExecutionMode = null;
     private Integer statelessInterOpThreads = null;
     private Integer statelessIntraOpThreads = null;
-    private GpuDevice gpuDevice = null;
+    private Integer gpuDevice = null;
 
     public OnnxModel(String name) {
         super(name);
@@ -114,9 +114,9 @@ public class OnnxModel extends DistributableResource {
         }
     }
 
-    public void setGpuDevice(int deviceNumber, boolean required) {
+    public void setGpuDevice(int deviceNumber) {
         if (deviceNumber >= 0) {
-            this.gpuDevice = new GpuDevice(deviceNumber, required);
+            this.gpuDevice = deviceNumber;
         }
     }
 
@@ -124,16 +124,8 @@ public class OnnxModel extends DistributableResource {
         return Optional.ofNullable(statelessIntraOpThreads);
     }
 
-    public Optional<GpuDevice> getGpuDevice() {
+    public Optional<Integer> getGpuDevice() {
         return Optional.ofNullable(gpuDevice);
     }
 
-    public record GpuDevice(int deviceNumber, boolean required) {
-
-        public GpuDevice {
-            if (deviceNumber < 0) throw new IllegalArgumentException("deviceNumber cannot be negative, got " + deviceNumber);
-        }
-
-    }
-
 }