Set TF-GNN's minimum TF version to TF 2.9

PiperOrigin-RevId: 505127996
tensorflow · Jan 27, 2023 · 6626331 · 6626331
1 parent e9a52ad
commit 6626331
Show file tree

Hide file tree

Showing 7 changed files with 17 additions and 6 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.11"
+  value: "2.11.*"
 }
 
 action {

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.11"
+  value: "2.11.*"
 }
 
 action {

diff --git a/kokoro/github/ubuntu/cpu/oldest/continuous.cfg b/kokoro/github/ubuntu/cpu/oldest/continuous.cfg
@@ -6,7 +6,11 @@ env_vars: {
 }
 env_vars: {
   key: "TF_VERSION"
-  value: "2.10"
+  value: "2.9.*"
+}
+env_vars: {
+  key: "TAG_FILTERS"
+  value: ",-tf_at_least_2_10"
 }
 
 action {

diff --git a/kokoro/github/ubuntu/cpu/oldest/presubmit.cfg b/kokoro/github/ubuntu/cpu/oldest/presubmit.cfg
@@ -6,7 +6,11 @@ env_vars: {
 }
 env_vars: {
   key: "TF_VERSION"
-  value: "2.10"
+  value: "2.9.*"
+}
+env_vars: {
+  key: "TAG_FILTERS"
+  value: ",-tf_at_least_2_10"
 }
 
 action {

diff --git a/setup.py b/setup.py
@@ -174,8 +174,8 @@ def get_version():
         'networkx',
         'pyarrow',
         # pylint:disable=g-line-too-long
-        'tensorflow>=2.10.0; platform_machine != "arm64" or platform_system != "Darwin"',
-        'tensorflow-macos>=2.10.0; platform_machine == "arm64" and platform_system == "Darwin"',
+        'tensorflow>=2.9.0; platform_machine != "arm64" or platform_system != "Darwin"',
+        'tensorflow-macos>=2.9.0; platform_machine == "arm64" and platform_system == "Darwin"',
         # pylint:enable=g-line-too-long
     ],
     python_requires='>=3.8,<4',

diff --git a/tensorflow_gnn/runner/tasks/BUILD b/tensorflow_gnn/runner/tasks/BUILD
@@ -19,6 +19,7 @@ py_strict_test(
     name = "classification_test",
     srcs = ["classification_test.py"],
     srcs_version = "PY3",
+    tags = ["tf_at_least_2_10"],
     deps = [
         ":classification",
         "//:expect_absl_installed",
@@ -43,6 +44,7 @@ py_strict_test(
     name = "regression_test",
     srcs = ["regression_test.py"],
     srcs_version = "PY3",
+    tags = ["tf_at_least_2_10"],
     deps = [
         ":regression",
         "//:expect_absl_installed",

diff --git a/tensorflow_gnn/runner/utils/BUILD b/tensorflow_gnn/runner/utils/BUILD
@@ -98,6 +98,7 @@ py_strict_test(
     name = "attribution_test",
     srcs = ["attribution_test.py"],
     srcs_version = "PY3",
+    tags = ["tf_at_least_2_10"],
     deps = [
         ":attribution",
         "//:expect_tensorflow_installed",