summaryrefslogtreecommitdiffstats
path: root/config-model
diff options
context:
space:
mode:
Diffstat (limited to 'config-model')
-rw-r--r--config-model/src/main/java/com/yahoo/vespa/model/container/component/Model.java4
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; }
}