disable testLarge for float16 when numpy 2.0 is used

tensorflower-gardener · tensorflower-gardener · commit 9e52ba9b69bb · 2024-09-10T15:21:22.000-07:00
PiperOrigin-RevId: 673106123
diff --git a/tensorflow/python/kernel_tests/control_flow/scan_ops_test.py b/tensorflow/python/kernel_tests/control_flow/scan_ops_test.py
@@ -153,6 +153,9 @@ def test6D(self):
   @test_util.disable_xla("b/123860949")  # The computation is constant folded
   def testLarge(self):
     for dtype in self.valid_dtypes:
+      if np.__version__ >= np.lib.NumpyVersion("2.0.0") and dtype == np.float16:
+        continue
+
       x = np.ones([1000000], dtype=dtype) / 1024
       self._compareAll(x, 0)
 
