Updated tests to prepare for NumPy2 upgrade

tf-transform-team · tfx-copybara · commit 57f428317581 · 2024-09-09T09:31:18.000-07:00
PiperOrigin-RevId: 672571140
diff --git a/tensorflow_transform/analyzers_test.py b/tensorflow_transform/analyzers_test.py
@@ -567,7 +567,7 @@ def testCombiner(self, combiner, batches, expected_outputs):
       self.assertEqual(output.dtype, expected_output.dtype)
       self.assertEqual(tensor_info.dtype, tf.as_dtype(expected_output.dtype))
 
-      self.assertAllEqual(output, expected_output)
+      self.assertAllClose(output, expected_output, rtol=1e-4, atol=1e-4)
 
   @test_case.named_parameters(
       {
diff --git a/tensorflow_transform/gaussianization_test.py b/tensorflow_transform/gaussianization_test.py
@@ -234,7 +234,7 @@ def test_compute_tukey_hh_params(
     output = gaussianization.compute_tukey_hh_params(l_skewness_and_kurtosis)
     self.assertEqual(output.dtype, expected_output.dtype)
     self.assertAllEqual(output.shape, expected_output.shape)
-    self.assertAllClose(output, expected_output)
+    self.assertAllClose(output, expected_output, rtol=1e-5, atol=1e-5)
 
   @test_case.named_parameters(*_LAMBERT_W_SCALAR_TESTS + _LAMBERT_W_ND_TESTS)
   def test_lambert_w(self, samples, expected_output):

Original file line number	Diff line number	Diff line change
`@@ -567,7 +567,7 @@ def testCombiner(self, combiner, batches, expected_outputs):`
`567`	`567`	`self.assertEqual(output.dtype, expected_output.dtype)`
`568`	`568`	`self.assertEqual(tensor_info.dtype, tf.as_dtype(expected_output.dtype))`
`569`	`569`
`570`		`- self.assertAllEqual(output, expected_output)`
	`570`	`+ self.assertAllClose(output, expected_output, rtol=1e-4, atol=1e-4)`
`571`	`571`
`572`	`572`	`@test_case.named_parameters(`
`573`	`573`	`{`