diff options
| author |  David Soergel <soergel@google.com> | 2017-09-28 11:55:38 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-09-28 12:09:27 -0700 |
| commit | 4abd3050b94f19157d919fef9aa515bbc4c01a93 (patch) | |
| tree | 9cfe19f90e59c22140cfa419c04db00b520871b3 /tensorflow/contrib/saved_model | |
| parent | f0c832dabc2531e56004a0d909fdb6437777e9c4 (diff) | |
Add more validation of serving signatures, both at creation and post hoc.
PiperOrigin-RevId: 170376578
Diffstat (limited to 'tensorflow/contrib/saved_model')
| -rw-r--r-- | tensorflow/contrib/saved_model/python/saved_model/signature_def_utils_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/saved_model/python/saved_model/signature_def_utils_test.py b/tensorflow/contrib/saved_model/python/saved_model/signature_def_utils_test.py index 282dd7dc3b..d2e14f73e4 100644 --- a/tensorflow/contrib/saved_model/python/saved_model/signature_def_utils_test.py +++ b/tensorflow/contrib/saved_model/python/saved_model/signature_def_utils_test.py @@ -94,7 +94,7 @@ class SignatureDefUtilsTest(test.TestCase): def testGetSignatureDefByKeyRegression(self): input1 = constant_op.constant("a", name="input-1") - output1 = constant_op.constant("b", name="output-1") + output1 = constant_op.constant(7.2, name="output-1") meta_graph_def = meta_graph_pb2.MetaGraphDef() self._add_to_signature_def_map(meta_graph_def, { @@ -123,7 +123,7 @@ class SignatureDefUtilsTest(test.TestCase): def testGetSignatureDefByKeyClassification(self): input1 = constant_op.constant("a", name="input-1") output1 = constant_op.constant("b", name="output-1") - output2 = constant_op.constant("c", name="output-2") + output2 = constant_op.constant(3.0, name="output-2") meta_graph_def = meta_graph_pb2.MetaGraphDef() self._add_to_signature_def_map(meta_graph_def, { |