experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Make padded_batch_test robust to the eq changes.

Browse Source 
PiperOrigin-RevId: 263840727
This commit is contained in:
Alexandre Passos 2019-08-16 13:59:24 -07:00 committed by TensorFlower Gardener
parent 1d825cff12
commit 76221c94dc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/data/kernel_tests/padded_batch_test.py
View File

@ -221,7 +221,7 @@ class PaddedBatchTest(test_base.DatasetTestBase, parameterized.TestCase):
TypeError, r'Padded shape .* must be a 1-D tensor '
r'of tf.int64 values, but its element type was float32.'):
_ = dataset_ops.Dataset.from_tensors([1, 2, 3]).padded_batch(
5, padded_shapes=constant_op.constant([1., 2., 3.]))
5, padded_shapes=constant_op.constant([1.5, 2., 3.]))
with self.assertRaisesRegexp(
ValueError, r'The padded shape \(1,\) is not compatible with the '