diff options
| author |  Alexandre Passos <apassos@google.com> | 2017-11-20 14:59:34 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-20 15:02:51 -0800 |
| commit | 775b496167bef8067d6c03f42809a96d565727f9 (patch) | |
| tree | 400c2cedd6632bc94315b7991f388d424ebebe9e /tensorflow/python/eager/backprop_test.py | |
| parent | 248d7c26c2a3f8d3f45b3498eff1d639e7cb0077 (diff) | |
Do not swallow exceptions in gradient functions in eager.
PiperOrigin-RevId: 176422128
Diffstat (limited to 'tensorflow/python/eager/backprop_test.py')
| -rw-r--r-- | tensorflow/python/eager/backprop_test.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/python/eager/backprop_test.py b/tensorflow/python/eager/backprop_test.py index 86c9cce3fd..e18ebba785 100644 --- a/tensorflow/python/eager/backprop_test.py +++ b/tensorflow/python/eager/backprop_test.py @@ -28,7 +28,6 @@ from tensorflow.python.eager import tape from tensorflow.python.eager import test from tensorflow.python.framework import constant_op from tensorflow.python.framework import dtypes -from tensorflow.python.framework import errors_impl from tensorflow.python.framework import ops from tensorflow.python.framework import tensor_shape from tensorflow.python.ops import array_ops @@ -111,7 +110,7 @@ class BackpropTest(test.TestCase): return x, grad # TODO(apassos) raise the right error here - with self.assertRaises(errors_impl.InternalError): + with self.assertRaises(RuntimeError): backprop.gradients_function(f)(constant_op.constant(1.0)) def testImplicitGradOverEmbeddingLookup(self): |