diff options
Diffstat (limited to 'model-evaluation')
-rw-r--r-- | model-evaluation/src/main/java/ai/vespa/models/evaluation/RankProfilesConfigImporter.java | 9 |
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); |