From 620fa97685630ced5f04d522a0462573ecd350f4 Mon Sep 17 00:00:00 2001 From: Tzu-Wei Sung Date: Tue, 8 Dec 2020 20:53:39 -0800 Subject: [PATCH] Use fixed number to test --- tensorflow/python/kernel_tests/unique_op_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensorflow/python/kernel_tests/unique_op_test.py b/tensorflow/python/kernel_tests/unique_op_test.py index 1b6ad50b371..ea52f76aca0 100644 --- a/tensorflow/python/kernel_tests/unique_op_test.py +++ b/tensorflow/python/kernel_tests/unique_op_test.py @@ -110,7 +110,7 @@ class UniqueTest(test.TestCase): @test_util.run_deprecated_v1 def testShapeInferenceV2(self): """Test shape inference.""" - x = np.random.randint(2, high=10, size=(3, 2, 1)) + x = np.arange(6).reshape(3, 2, 1) _, idx = gen_array_ops.unique_v2(x, axis=[0]) self.assertEqual(idx.shape.as_list(), [3]) _, idx = gen_array_ops.unique_v2(x, axis=[1])