diff --git a/tensorflow/python/kernel_tests/unique_op_test.py b/tensorflow/python/kernel_tests/unique_op_test.py
index 0a0306e76f1852..448c9d6df55bbf 100644
--- a/tensorflow/python/kernel_tests/unique_op_test.py
+++ b/tensorflow/python/kernel_tests/unique_op_test.py
@@ -282,7 +282,9 @@ def testFloat(self):
     tf_y, tf_idx, tf_count = self.evaluate([y, idx, count])
 
     self.assertEqual(len(x), len(tf_idx))
-    self.assertEqual(len(tf_y), len(np.unique(x)))
+    # TODO(b/202197513): numpy>=1.20.0 makes np.unique() treat np.nan as equal
+    # so len(np.unique(x)) == 3. So this no longer can be tested this way.
+    # self.assertEqual(len(tf_y), len(np.unique(x)))
     for i in range(len(x)):
       if np.isnan(x[i]):
         self.assertTrue(np.isnan(tf_y[tf_idx[i]]))