Re-enables edge sampling tests.

PiperOrigin-RevId: 548669345
tensorflow · Jul 17, 2023 · a3f56f9 · a3f56f9
1 parent 83795b5
commit a3f56f9
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/tensorflow_gnn/experimental/sampler/beam/edge_samplers_test.py b/tensorflow_gnn/experimental/sampler/beam/edge_samplers_test.py
@@ -54,8 +54,6 @@ def sampling_results_equal(self, expected, actual) -> bool:
 class EdgeSamplersTest(ExecutorTestBase):
 
   def test_sampling(self):
-    # TODO(b/291057185): Re-enable when beam bug is fixed or worked around.
-    self.skipTest('Test started failing due to a beam problem with GroupByKey.')
     edges = {1: 2, 2: 3}
     feats = {
         1: text_format.Merge(