aboutsummaryrefslogtreecommitdiffstats
path: root/model-evaluation
diff options
context:
space:
mode:
Diffstat (limited to 'model-evaluation')
-rw-r--r--model-evaluation/src/main/java/ai/vespa/models/evaluation/RankProfilesConfigImporter.java9
1 files changed, 6 insertions, 3 deletions
diff --git a/model-evaluation/src/main/java/ai/vespa/models/evaluation/RankProfilesConfigImporter.java b/model-evaluation/src/main/java/ai/vespa/models/evaluation/RankProfilesConfigImporter.java
index 9877dd69e83..e8aae24ca9e 100644
--- a/model-evaluation/src/main/java/ai/vespa/models/evaluation/RankProfilesConfigImporter.java
+++ b/model-evaluation/src/main/java/ai/vespa/models/evaluation/RankProfilesConfigImporter.java
@@ -232,13 +232,16 @@ public class RankProfilesConfigImporter {
protected Tensor readTensorFromFile(String name, TensorType type, FileReference fileReference) {
try {
File file = fileAcquirer.waitFor(fileReference, 7, TimeUnit.DAYS);
- if (file.getName().endsWith(".tbf"))
+ if (file.getName().endsWith(".tbf")) {
return TypedBinaryFormat.decode(Optional.of(type),
GrowableByteBuffer.wrap(IOUtils.readFileBytes(file)));
- else
+ } else if (file.getName().endsWith(".json")) {
+ return com.yahoo.tensor.serialization.JsonFormat.decode(type, IOUtils.readFileBytes(file));
+ } else {
throw new IllegalArgumentException("Constant files on other formats than .tbf are not supported, got " +
file + " for constant " + name);
- // TODO: Support json and json.lz4
+ }
+ // TODO: Support json.lz4
}
catch (InterruptedException e) {
throw new IllegalStateException("Gave up waiting for constant " + name);