diff options
author | 2017-09-28 11:55:38 -0700 | |
---|---|---|
committer | 2017-09-28 12:01:58 -0700 | |
commit | 996b0342879af43de1bf4071190b90ff7309428a (patch) | |
tree | 9cfe19f90e59c22140cfa419c04db00b520871b3 /tensorflow/contrib/saved_model | |
parent | 0254d0d31337724db911c89609336afd60e8192d (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, { |