remove .numpy which failed in gpu/distributed env

tensorflow · Apr 29, 2024 · 1363734 · 1363734
1 parent eabccef
commit 1363734
Showing 1 changed file with 4 additions and 5 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
@@ -19,9 +19,8 @@ def test_fill_empty_rows(self):
     expected_empty = tf.constant([False, True, False, True, False])
 
     self.assertTrue(
-        tf.reduce_all(filled_ragged_tensor.to_tensor() ==
-                      expected_filled.to_tensor()).numpy(),
+        tf.reduce_all(
+            filled_ragged_tensor.to_tensor() == expected_filled.to_tensor()),
         "Filled tensors do not match")
-    self.assertTrue(
-        tf.reduce_all(is_row_empty == expected_empty).numpy(),
-        "Empty row flags do not match")
+    self.assertTrue(tf.reduce_all(is_row_empty == expected_empty),
+                    "Empty row flags do not match")