From 320e78201040356cb55e43c22bbe26d6e4302330 Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Thu, 25 Jul 2024 08:03:05 -0700 Subject: [PATCH] [numpy] Fix users of NumPy APIs that are removed in NumPy 2.0. This change migrates users of APIs removed in NumPy 2.0 to their recommended replacements (https://numpy.org/devdocs/numpy_2_0_migration_guide.html). PiperOrigin-RevId: 655956468 --- tensorflow_graphics/util/tests/asserts_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow_graphics/util/tests/asserts_test.py b/tensorflow_graphics/util/tests/asserts_test.py index 5ba9b8a76..563f3a84a 100644 --- a/tensorflow_graphics/util/tests/asserts_test.py +++ b/tensorflow_graphics/util/tests/asserts_test.py @@ -347,7 +347,7 @@ def test_select_eps_for_addition(self, dtype): with self.assertRaises(tf.errors.InvalidArgumentError): self.evaluate(tf.debugging.assert_equal(a, a + eps)) - @parameterized.parameters((np.NaN,), (np.inf,)) + @parameterized.parameters((np.nan,), (np.inf,)) @flagsaver.flagsaver(tfg_add_asserts_to_graph=False) def test_assert_no_infs_or_nans_passthrough(self, value): """Checks that the assert is a passthrough when the flag is False.""" @@ -357,7 +357,7 @@ def test_assert_no_infs_or_nans_passthrough(self, value): self.assertIs(vector_input, vector_output) - @parameterized.parameters((np.NaN,), (np.inf,)) + @parameterized.parameters((np.nan,), (np.inf,)) def test_assert_no_infs_or_nans_raises_exception_for_nan(self, value): """Checks that the assert works for `Inf` or `NaN` values.""" vector_input = (value,)