diff options
author | Henning Baldersheim <balder@yahoo-inc.com> | 2021-08-17 06:49:12 +0200 |
---|---|---|
committer | Henning Baldersheim <balder@yahoo-inc.com> | 2021-08-17 06:49:12 +0200 |
commit | acec3af18f52b1713f9231a32a2203ebd6910f1c (patch) | |
tree | fa68e3b75233b3178f7787351adf6ca702b9631f /config-model/src/main/java/com/yahoo/searchdefinition/OnnxModels.java | |
parent | 08c2567e0d63a9cffbd18713f5c2b31cf9cf7366 (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.java | 5 |
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(); |