diff options
| author | Jon Bratseth <bratseth@gmail.com> | 2022-11-09 16:15:20 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-11-09 16:15:20 +0100 |
| commit | 606bc6b517e09bba9f36a4ef6506a588d1254bd4 (patch) | |
| tree | 207c9dc61d3207e4630e73dc51ac5bdff1ac6d40 /model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/TestableModel.java | |
| parent | df9c6df4d0282a765fd9831eeb3bfd8ac5e38faa (diff) | |
| parent | fec896323cb8af7e20c5b7a0ca3fbeba7457b4d2 (diff) | |
Merge pull request #24813 from vespa-engine/revert-24773-revert-24760-balder/model-importing-code-in-config-model-3
Revert "Revert "Balder/model importing code in config model [run-systemtest]""
Diffstat (limited to 'model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/TestableModel.java')
| -rw-r--r-- | model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/TestableModel.java | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/TestableModel.java b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/TestableModel.java index 5d13697df06..f78150d9875 100644 --- a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/TestableModel.java +++ b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/onnx/TestableModel.java @@ -63,8 +63,8 @@ public class TestableModel { static Context contextFrom(ImportedModel result) { TestableModelContext context = new TestableModelContext(); - result.largeConstants().forEach((name, tensor) -> context.put("constant(" + name + ")", new TensorValue(Tensor.from(tensor)))); - result.smallConstants().forEach((name, tensor) -> context.put("constant(" + name + ")", new TensorValue(Tensor.from(tensor)))); + result.largeConstantTensors().forEach((name, tensor) -> context.put("constant(" + name + ")", new TensorValue(tensor))); + result.smallConstantTensors().forEach((name, tensor) -> context.put("constant(" + name + ")", new TensorValue(tensor))); return context; } |