aboutsummaryrefslogtreecommitdiffstats
path: root/config-model/src/main/java/com/yahoo/searchdefinition/OnnxModels.java
diff options
context:
space:
mode:
authorHenning Baldersheim <balder@yahoo-inc.com>2021-08-17 06:49:12 +0200
committerHenning Baldersheim <balder@yahoo-inc.com>2021-08-17 06:49:12 +0200
commitacec3af18f52b1713f9231a32a2203ebd6910f1c (patch)
treefa68e3b75233b3178f7787351adf6ca702b9631f /config-model/src/main/java/com/yahoo/searchdefinition/OnnxModels.java
parent08c2567e0d63a9cffbd18713f5c2b31cf9cf7366 (diff)
Wire in fileregistry to RankingConstants and OnnxModels.
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.java5
1 files changed, 5 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 60733a4f5ba..96ee8c0c2f4 100644
--- a/config-model/src/main/java/com/yahoo/searchdefinition/OnnxModels.java
+++ b/config-model/src/main/java/com/yahoo/searchdefinition/OnnxModels.java
@@ -1,6 +1,7 @@
// Copyright Verizon Media. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.searchdefinition;
+import com.yahoo.config.application.api.FileRegistry;
import com.yahoo.vespa.model.AbstractService;
import java.util.Collection;
@@ -16,7 +17,11 @@ import java.util.Map;
public class OnnxModels {
private final Map<String, OnnxModel> models = new HashMap<>();
+ private final FileRegistry fileRegistry;
+ public OnnxModels(FileRegistry fileRegistry) {
+ this.fileRegistry = fileRegistry;
+ }
public void add(OnnxModel model) {
model.validate();
String name = model.getName();