diff options
author | 2018-01-23 15:02:46 -0800 | |
---|---|---|
committer | 2018-01-23 15:08:42 -0800 | |
commit | 68a9ee1d4e041d7690a949718b2651b035a6bfad (patch) | |
tree | a93474b55a32e0ab467b01e76ee481cbf319fe8a /tensorflow/contrib/labeled_tensor | |
parent | 67ed32d35148f382de92c9383347400875eaba40 (diff) |
LabeledTensor: don't test Tensor.shape object, test value instead.
With the C API enabled, a new TensorShape object may be generated
every Tensor.get_shape call.
PiperOrigin-RevId: 182998609
Diffstat (limited to 'tensorflow/contrib/labeled_tensor')
-rw-r--r-- | tensorflow/contrib/labeled_tensor/python/ops/core_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/labeled_tensor/python/ops/core_test.py b/tensorflow/contrib/labeled_tensor/python/ops/core_test.py index 1f4a3ef568..e70b492374 100644 --- a/tensorflow/contrib/labeled_tensor/python/ops/core_test.py +++ b/tensorflow/contrib/labeled_tensor/python/ops/core_test.py @@ -225,7 +225,7 @@ class LabeledTensorTest(test_util.Base): tensor = array_ops.placeholder(dtypes.string, [None]) actual = core.LabeledTensor(tensor, ['x']) self.assertIsNone(actual.axes['x'].size) - self.assertIs(actual.axes['x'].value, tensor.get_shape()[0]) + self.assertIsNone(actual.axes['x'].value.value) def test_eq(self): self.assertEqual(self.lt, self.lt) |