aboutsummaryrefslogtreecommitdiffstats
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.java6
1 files changed, 5 insertions, 1 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 1cc33664e8c..60733a4f5ba 100644
--- a/config-model/src/main/java/com/yahoo/searchdefinition/OnnxModels.java
+++ b/config-model/src/main/java/com/yahoo/searchdefinition/OnnxModels.java
@@ -9,7 +9,7 @@ import java.util.HashMap;
import java.util.Map;
/**
- * ONNX models tied to a search definition.
+ * ONNX models tied to a search definition or global.
*
* @author lesters
*/
@@ -23,6 +23,10 @@ public class OnnxModels {
models.put(name, model);
}
+ public void add(Map<String, OnnxModel> models) {
+ models.values().forEach(this::add);
+ }
+
public OnnxModel get(String name) {
return models.get(name);
}