diff options
Diffstat (limited to 'config-model')
-rw-r--r-- | config-model/src/main/java/com/yahoo/vespa/model/container/component/Model.java | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/config-model/src/main/java/com/yahoo/vespa/model/container/component/Model.java b/config-model/src/main/java/com/yahoo/vespa/model/container/component/Model.java index 084258ce1a6..76d93c38aee 100644 --- a/config-model/src/main/java/com/yahoo/vespa/model/container/component/Model.java +++ b/config-model/src/main/java/com/yahoo/vespa/model/container/component/Model.java @@ -55,13 +55,15 @@ class Model { } void registerOnnxModelCost(ApplicationContainerCluster c) { + var resolvedUrl = resolvedUrl().orElse(null); if (file != null) c.onnxModelCost().registerModel(file); - else c.onnxModelCost().registerModel(url); + else if (resolvedUrl != null) c.onnxModelCost().registerModel(resolvedUrl); } String name() { return paramName; } Optional<String> modelId() { return Optional.ofNullable(modelId); } Optional<URI> url() { return Optional.ofNullable(url); } + Optional<URI> resolvedUrl() { return ref.url().map(u -> URI.create(u.value())); } Optional<ApplicationFile> file() { return Optional.ofNullable(file); } ModelReference modelReference() { return ref; } } |