Partially re-enable some tests for TF2.9.

PiperOrigin-RevId: 507798778

tensorflow_gnn/runner/utils/BUILD

@@ -98,7 +98,6 @@ py_strict_test(
     name = "attribution_test",
     srcs = ["attribution_test.py"],
     srcs_version = "PY3",
-    tags = ["tf_at_least_2_10"],
     deps = [
         ":attribution",
         "//:expect_tensorflow_installed",

tensorflow_gnn/runner/utils/attribution_test.py

@@ -52,6 +52,10 @@ class AttributionTest(tf.test.TestCase):
   def test_counterfactual_random(self):
     counterfactual = attribution.counterfactual(self.gt, random=True, seed=8191)
 
+    # TODO(b/266817638): Remove when fixed.
+    if tf.__version__.startswith("2.9."):
+      self.skipTest("Bad test: exepected values depend on TF2.10+ RNG seeds")
+
     self.assertAllEqual(
         counterfactual.context.features["h"],
         as_tensor((0.49280962, 0.466383)))
@@ -95,6 +99,10 @@ def test_subtract_graph_features(self):
                 }
             }))
 
+    # TODO(b/266817638): Remove when fixed.
+    if tf.__version__.startswith("2.9."):
+      self.skipTest("Bad test: exepected values depend on TF2.10+ RNG seeds")
+
     self.assertAllClose(
         deltas.context.features["h"],
         as_tensor((.514 - .4, .433 - .8)))
@@ -116,6 +124,10 @@ def test_interpolate(self):
 
     self.assertLen(interpolations, 4)
 
+    # TODO(b/266817638): Remove when fixed.
+    if tf.__version__.startswith("2.9."):
+      self.skipTest("Bad test: exepected values depend on TF2.10+ RNG seeds")
+
     # Interpolation 0
     self.assertAllEqual(
         interpolations[0].context.features["h"],
@@ -179,6 +191,10 @@ def test_interpolate(self):
   def test_sum_graph_features(self):
     summation = attribution.sum_graph_features((self.gt,) * 4)
 
+    # TODO(b/266817638): Remove when fixed.
+    if tf.__version__.startswith("2.9."):
+      self.skipTest("Bad test: exepected values depend on TF2.10+ RNG seeds")
+
     self.assertAllEqual(
         summation.context.features["h"],
         as_tensor((.514 * 4, .433 * 4)))
@@ -258,6 +274,10 @@ def test_integrated_gradients_exporter(self):
     outputs = saved_model.signatures["integrated_gradients"](**kwargs)
     gt = outputs["output"]
 
+    # TODO(b/266817638): Remove when fixed.
+    if tf.__version__.startswith("2.9."):
+      self.skipTest("Bad test: exepected values depend on TF2.10+ RNG seeds")
+
     # The above GNN passes a single message over the only edge type before
     # collecting a seed node for activations. The above graph is a line:
     # seed --weight 0--> node 1 --weight 1--> node 2.