Internal change.

PiperOrigin-RevId: 689715583
tensorflow · Oct 25, 2024 · 93cd2be · 93cd2be
1 parent 7befc1a
commit 93cd2be
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/kokoro/github/ubuntu/cpu/newest_stable/continuous.cfg b/kokoro/github/ubuntu/cpu/newest_stable/continuous.cfg
@@ -6,7 +6,7 @@ env_vars: {
 }
 env_vars: {
   key: "TF_VERSION"
-  value: "2.17.*"
+  value: "2.18.*"
 }
 env_vars: {
   key: "TF_USE_LEGACY_KERAS"

diff --git a/kokoro/github/ubuntu/cpu/newest_stable/presubmit.cfg b/kokoro/github/ubuntu/cpu/newest_stable/presubmit.cfg
@@ -6,7 +6,7 @@ env_vars: {
 }
 env_vars: {
   key: "TF_VERSION"
-  value: "2.17.*"
+  value: "2.18.*"
 }
 env_vars: {
   key: "TF_USE_LEGACY_KERAS"

diff --git a/tensorflow_gnn/experimental/sampler/beam/executor_lib_test.py b/tensorflow_gnn/experimental/sampler/beam/executor_lib_test.py
@@ -214,6 +214,7 @@ def test_any_composite(self):
         tf.__version__.startswith('2.15')
         or tf.__version__.startswith('2.16')
         or tf.__version__.startswith('2.17')
+        or tf.__version__.startswith('2.18')
         or '-dev' in tf.__version__  # tf-nightly
     ):
       self.skipTest('Fails with tf internal error since tf 2.15')