diff options
Diffstat (limited to 'searchlib')
-rw-r--r-- | searchlib/src/test/java/com/yahoo/searchlib/rankingexpression/integration/ml/BatchNormImportTestCase.java | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/searchlib/src/test/java/com/yahoo/searchlib/rankingexpression/integration/ml/BatchNormImportTestCase.java b/searchlib/src/test/java/com/yahoo/searchlib/rankingexpression/integration/ml/BatchNormImportTestCase.java index 6eb13ebbd6c..bf9684082f4 100644 --- a/searchlib/src/test/java/com/yahoo/searchlib/rankingexpression/integration/ml/BatchNormImportTestCase.java +++ b/searchlib/src/test/java/com/yahoo/searchlib/rankingexpression/integration/ml/BatchNormImportTestCase.java @@ -23,9 +23,7 @@ public class BatchNormImportTestCase { RankingExpression output = signature.outputExpression("y"); assertNotNull(output); assertEquals("dnn/batch_normalization_3/batchnorm/add_1", output.getName()); -// model.assertEqualResult("X", output.getName()); - - model.assertEqualResult("X", "dnn/outputs/BiasAdd"); + model.assertEqualResult("X", output.getName()); } } |