summaryrefslogtreecommitdiffstats
path: root/config-model/src/main/java/com/yahoo/searchdefinition/OnnxModels.java
diff options
context:
space:
mode:
Diffstat (limited to 'config-model/src/main/java/com/yahoo/searchdefinition/OnnxModels.java')
-rw-r--r--config-model/src/main/java/com/yahoo/searchdefinition/OnnxModels.java4
1 files changed, 4 insertions, 0 deletions
diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/OnnxModels.java b/config-model/src/main/java/com/yahoo/searchdefinition/OnnxModels.java
index 87663ac79a3..1cc33664e8c 100644
--- a/config-model/src/main/java/com/yahoo/searchdefinition/OnnxModels.java
+++ b/config-model/src/main/java/com/yahoo/searchdefinition/OnnxModels.java
@@ -27,6 +27,10 @@ public class OnnxModels {
return models.get(name);
}
+ public boolean has(String name) {
+ return models.containsKey(name);
+ }
+
public Map<String, OnnxModel> asMap() {
return Collections.unmodifiableMap(models);
}