From 76221c94dc4561be9484d8879643f1d4a82f9058 Mon Sep 17 00:00:00 2001 From: Alexandre Passos Date: Fri, 16 Aug 2019 13:59:24 -0700 Subject: [PATCH] Make padded_batch_test robust to the eq changes. PiperOrigin-RevId: 263840727 --- tensorflow/python/data/kernel_tests/padded_batch_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/python/data/kernel_tests/padded_batch_test.py b/tensorflow/python/data/kernel_tests/padded_batch_test.py index b54749002cc..39339c0063a 100644 --- a/tensorflow/python/data/kernel_tests/padded_batch_test.py +++ b/tensorflow/python/data/kernel_tests/padded_batch_test.py @@ -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 '