diff options
author | Jon Bratseth <bratseth@verizonmedia.com> | 2019-07-03 09:47:49 -0700 |
---|---|---|
committer | Jon Bratseth <bratseth@verizonmedia.com> | 2019-07-03 09:47:49 -0700 |
commit | 0ce6fa7cbdf71fd39cb5bb18accfa84a20e7e120 (patch) | |
tree | 0afa3697a10fad159883a402a2f367ee8175c027 /model-integration/src/test | |
parent | fdc6a48fe913de5f5a84c6eb42123543c1d2ee46 (diff) |
Inject rename to relax constraint proof of concept
Diffstat (limited to 'model-integration/src/test')
3 files changed, 4 insertions, 4 deletions
diff --git a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/DimensionRenamerTest.java b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/DimensionRenamerTest.java index ee282c7d988..793258868ee 100644 --- a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/DimensionRenamerTest.java +++ b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/DimensionRenamerTest.java @@ -9,7 +9,7 @@ public class DimensionRenamerTest { @Test public void testMnistRenaming() { - DimensionRenamer renamer = new DimensionRenamer(); + DimensionRenamer renamer = new DimensionRenamer(new IntermediateGraph("test")); renamer.addDimension("first_dimension_of_x"); renamer.addDimension("second_dimension_of_x"); diff --git a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/tensorflow/Issue9662TestCase.java b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/tensorflow/Issue9662TestCase.java index be0ab4b894a..6500a380190 100644 --- a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/tensorflow/Issue9662TestCase.java +++ b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/tensorflow/Issue9662TestCase.java @@ -20,9 +20,9 @@ public class Issue9662TestCase { Assert.assertEquals("Should have no skipped outputs", 0, model.get().signature("serving_default").skippedOutputs().size()); - ImportedMlFunction output = signature.outputFunction("y", "y"); + ImportedMlFunction output = signature.outputFunction("output", "output"); assertNotNull(output); - model.assertEqualResultSum("input", "dnn/outputs/add", 0.00001); + model.assertEqualResultSum("input_embedding_user_guid", "dense_out/MatMul", 0.00001); } } diff --git a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/tensorflow/TestableTensorFlowModel.java b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/tensorflow/TestableTensorFlowModel.java index 9d2f8cf0692..75fa2ed7933 100644 --- a/model-integration/src/test/java/ai/vespa/rankingexpression/importer/tensorflow/TestableTensorFlowModel.java +++ b/model-integration/src/test/java/ai/vespa/rankingexpression/importer/tensorflow/TestableTensorFlowModel.java @@ -49,7 +49,7 @@ public class TestableTensorFlowModel { public ImportedModel get() { return model; } - /** Compare that summing the tensors produce the same result to within some tolerance delta */ + /** Compare that computing the expressions produce the same result to within some tolerance delta */ public void assertEqualResultSum(String inputName, String operationName, double delta) { Tensor tfResult = tensorFlowExecute(tensorFlowModel, inputName, operationName); Context context = contextFrom(model); |