From 3e7ba8944b872e755f63e0ebeee649ad97adb9ec Mon Sep 17 00:00:00 2001 From: Hye Soo Yang Date: Wed, 27 May 2020 22:55:00 -0700 Subject: [PATCH] Explicitly specify input array `dtype` to TensorFlow's `assertAllEqual()` test util function. PiperOrigin-RevId: 313530667 Change-Id: Ia251459edaeb89318d012cf0ff54c9b43a56fe49 --- tensorflow/python/kernel_tests/slice_op_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/python/kernel_tests/slice_op_test.py b/tensorflow/python/kernel_tests/slice_op_test.py index 15a340ef270..b53147552c3 100644 --- a/tensorflow/python/kernel_tests/slice_op_test.py +++ b/tensorflow/python/kernel_tests/slice_op_test.py @@ -247,8 +247,8 @@ class SliceTest(test.TestCase): slice_t = array_ops.slice(a, [0, 0], [2, 2]) slice2_t = a[:2, :2] slice_val, slice2_val = self.evaluate([slice_t, slice2_t]) - self.assertAllEqual(slice_val, inp[:2, :2]) - self.assertAllEqual(slice2_val, inp[:2, :2]) + self.assertAllEqual(slice_val, np.array(inp[:2, :2], dtype=np.float32)) + self.assertAllEqual(slice2_val, np.array(inp[:2, :2], dtype=np.float32)) self.assertEqual(slice_val.shape, slice_t.get_shape()) self.assertEqual(slice2_val.shape, slice2_t.get_shape())