remove .numpy which failed in gpu/distributed env

tensorflow · Apr 29, 2024 · 52d8085 · 52d8085
1 parent eabccef
commit 52d8085
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...ow_recommenders_addons/dynamic_embedding/python/kernel_tests/ragged_embedding_ops_test.py b/...ow_recommenders_addons/dynamic_embedding/python/kernel_tests/ragged_embedding_ops_test.py
@@ -20,8 +20,8 @@ def test_fill_empty_rows(self):
 
     self.assertTrue(
         tf.reduce_all(filled_ragged_tensor.to_tensor() ==
-                      expected_filled.to_tensor()).numpy(),
+                      expected_filled.to_tensor()),
         "Filled tensors do not match")
     self.assertTrue(
-        tf.reduce_all(is_row_empty == expected_empty).numpy(),
+        tf.reduce_all(is_row_empty == expected_empty),
         "Empty row flags do not match")