Migrate package metadata to pyproject.toml.

PiperOrigin-RevId: 829082204

Author: sgarciagoogle

.bazelversion

@@ -0,0 +1,2 @@
+7.7.0
+# TODO: b/390391579 - Upgrade once bazel 8+ works.

BUILD

@@ -3,6 +3,7 @@ py_library(
     # This is a dummy rule used as a absl dependency in open-source.
     # We expect absl to already be installed on the system, e.g. via
     # `pip install absl`
+    strict_deps = False,
     visibility = ["//visibility:public"],
     deps = [],
 )
@@ -12,6 +13,7 @@ py_library(
     # This is a dummy rule used as a absl dependency in open-source.
     # We expect absl to already be installed on the system, e.g. via
     # `pip install absl`
+    strict_deps = False,
     visibility = ["//visibility:public"],
     deps = [],
 )
@@ -21,6 +23,7 @@ py_library(
     # This is a dummy rule used as a absl dependency in open-source.
     # We expect absl to already be installed on the system, e.g. via
     # `pip install absl`
+    strict_deps = False,
     visibility = ["//visibility:public"],
     deps = [],
 )
@@ -30,6 +33,7 @@ py_library(
     # This is a dummy rule used as a numpy dependency in open-source.
     # We expect numpy to already be installed on the system, e.g. via
     # `pip install numpy`
+    strict_deps = False,
     visibility = ["//visibility:public"],
     deps = [],
 )
@@ -39,6 +43,7 @@ py_library(
     # This is a dummy rule used as a tensorflow dependency in open-source.
     # We expect tensorflow to already be installed on the system, e.g. via
     # `pip install tensorflow`
+    strict_deps = False,
     visibility = ["//visibility:public"],
     deps = [],
 )
@@ -48,6 +53,7 @@ py_library(
     # This is a dummy rule used as a ai-edge-litert dependency in open-source.
     # We expect ai-edge-litert to already be installed on the system, e.g. via
     # `pip install ai-edge-litert`
+    strict_deps = False,
     visibility = ["//visibility:public"],
     deps = [],
 )
@@ -57,6 +63,7 @@ py_library(
     # This is a dummy rule used as a tensorflow dependency in open-source.
     # We expect tensorflow to already be installed on the system, e.g. via
     # `pip install tensorflow`
+    strict_deps = False,
     visibility = ["//visibility:public"],
     deps = [],
 )
@@ -66,6 +73,7 @@ py_library(
     # This is a dummy rule used as a dependency on vizier.service.pyvizier in open-source.
     # We expect Vizier to already be installed on the system, e.g. via
     # `pip install google-vizier` and extra steps if needed (b/254806045).
+    strict_deps = False,
     visibility = ["//visibility:public"],
     deps = [],
 )
@@ -75,6 +83,7 @@ py_library(
     # This is a dummy rule used as a mock dependency in open-source tests.
     # We expect mock to already be installed on the system, e.g. via
     # `pip install mock`
+    strict_deps = False,
     visibility = ["//visibility:public"],
     deps = [],
 )

kokoro/github/ubuntu/cpu/build.sh

@@ -32,16 +32,18 @@ cd "${KOKORO_ARTIFACTS_DIR}/github/gnn/"
 
 PIP_TEST_PREFIX=bazel_pip
 
-python -m venv venv
-source venv/bin/activate
+python -m venv build_venv
+source build_venv/bin/activate
 
 # Check the python version
 python --version
-python3 --version
 
 # update pip
 pip install --upgrade pip
 
+# Install build
+pip install build
+
 TEST_ROOT=$(pwd)/${PIP_TEST_PREFIX}
 rm -rf "$TEST_ROOT"
 mkdir -p "$TEST_ROOT"
@@ -55,25 +57,34 @@ if [[ -n "${USE_BAZEL_VERSION}" && $(bazel --version) != *${USE_BAZEL_VERSION}*
 fi
 
 bazel clean
-pip install -r requirements-dev.txt --progress-bar off
-pip install tf-keras-nightly tf-nightly --progress-bar off --upgrade
-# We need to remove the dependency on tensorflow to test nightly
-# The dependencies will be provided by tf-nightly
-perl  -i -lpe '$k+= s/tensorflow>=2\.[0-9]+\.[0-9]+(,<=?[0-9.]+)?;/tf-nightly;/g; END{exit($k != 1)}' setup.py
-# tf-nightly 2.21 lifted the protobuf version to what google-vizier 0.1.24 wants
-# but what google-vizier 0.1.21 together with apache-beam does not allow.
-# TODO: b/482003228 - Replace this patch by a proper dependency cleanup.
-perl  -i -lpe '$k+= s/google-vizier>=0.0.13,!=0.1.23,!=0.1.24/google-vizier>=0.0.13,!=0.1.23/g; END{exit($k != 1)}' setup.py
-python3 setup.py bdist_wheel
-pip uninstall -y tensorflow_gnn
+pip install --group test-nightly --progress-bar off --upgrade
+python3 -m build --wheel
+deactivate
+
+# Start the test environment.
+python3 -m venv test_venv
+source test_venv/bin/activate
+
+# Check the python version
+python --version
+
+pip install --upgrade pip
 pip install dist/tensorflow_gnn-*.whl
+pip uninstall -y tensorflow tf-keras ai-edge-litert
+pip install --group test-nightly --progress-bar off --upgrade
 
 echo "Final packages after all pip commands:"
 pip list
 
 # Check that tf-nightly is installed but tensorflow is not
 # Also check that tf-keras-nightly is installed.
-pip freeze | grep -q tf-nightly= && ! pip freeze | grep -q tensorflow=
-pip freeze | grep -q tf-keras-nightly= && ! pip freeze | grep -q tf-keras=
+if [[ $(pip freeze | grep -q tf_nightly=; echo $?) -eq 0 && $(pip freeze | grep -q tensorflow=; echo $?) -eq 0 ]]; then
+  echo "Found tensorflow and tf_nightly in the environment."
+  exit 1
+fi
+if [[ $(pip freeze | grep -q tf_keras-nightly=; echo $?) -eq 0 && $(pip freeze | grep -q tf_keras=; echo $?) -eq 0 ]]; then
+  echo "Found tf_keras and tf_keras-nightly in the environment."
+  exit 1
+fi
 # The env variable is needed to ensure that TF keras still behaves like keras 2
 bazel test --test_env="TF_USE_LEGACY_KERAS=1" --build_tag_filters="${tag_filters}" --test_tag_filters="${tag_filters}" --test_output=errors --verbose_failures=true --build_tests_only --define=no_tfgnn_py_deps=true --keep_going --experimental_repo_remote_exec //bazel_pip/tensorflow_gnn/...

kokoro/github/ubuntu/cpu/build_versioned.sh

@@ -19,6 +19,18 @@ set -x
 PYENV_ROOT="/home/kbuilder/.pyenv"
 PYTHON_VERSION=${PYTHON_VERSION:-"3.11"}
 
+function force_tensorflow_version() {
+  if [[ -z "${TF_VERSION}" ]]; then
+    echo "TF_VERSION is not set. Not forcing tensorflow version."
+    return
+  fi
+
+  pip install tensorflow=="${TF_VERSION}" --progress-bar off --upgrade
+  if [[ "$TF_USE_LEGACY_KERAS" == 1 ]]; then
+    pip install tf-keras=="${TF_VERSION}" --progress-bar off --upgrade
+  fi
+}
+
 echo "Installing pyenv.."
 git clone https://github.com/pyenv/pyenv.git "$PYENV_ROOT"
 export PATH="/home/kbuilder/.local/bin:$PYENV_ROOT/bin:$PATH"
@@ -32,16 +44,18 @@ cd "${KOKORO_ARTIFACTS_DIR}/github/gnn/"
 
 PIP_TEST_PREFIX=bazel_pip
 
-python -m venv venv
-source venv/bin/activate
+python -m venv build_venv
+source build_venv/bin/activate
 
 # Debug messages to indicate the python version
 python --version
-python3 --version
 
 # update pip
 pip install --upgrade pip
 
+# Install build
+pip install build
+
 TEST_ROOT=$(pwd)/${PIP_TEST_PREFIX}
 rm -rf "$TEST_ROOT"
 mkdir -p "$TEST_ROOT"
@@ -62,13 +76,27 @@ if [[ -n "${USE_BAZEL_VERSION}" && $(bazel --version) != *${USE_BAZEL_VERSION}*
 fi
 
 bazel clean
-pip install -r requirements-dev.txt --progress-bar off
-pip install tensorflow=="${TF_VERSION}" --progress-bar off --upgrade
+force_tensorflow_version
+python3 -m build --wheel
+deactivate
+
+# Start the test environment.
+python3 -m venv test_venv
+source test_venv/bin/activate
+
+# Check the python version
+python --version
+
+# update pip
+pip install --upgrade pip
+force_tensorflow_version
+
 if [[ "$TF_USE_LEGACY_KERAS" == 1 ]]; then
-  pip install tf-keras=="${TF_VERSION}" --progress-bar off --upgrade
+  pip install --group test-tf216plus --progress-bar off --upgrade
+else
+  pip install --group test-pre-tf216 --progress-bar off --upgrade
 fi
-python3 setup.py bdist_wheel
-pip uninstall -y tensorflow_gnn
+
 pip install dist/tensorflow_gnn-*.whl
 
 echo "Final packages after all pip commands:"

kokoro/github/ubuntu/cpu/continuous.cfg

@@ -1,9 +1,5 @@
 build_file: "gnn/kokoro/github/ubuntu/cpu/build.sh"
 
-env_vars: {
-  key: "USE_BAZEL_VERSION"
-  value: "7.4.1"  # TODO - b/390391579: Unpin once bazel 8 works.
-}
 env_vars: {
   key: "PYTHON_VERSION"
   value: "3.11"

kokoro/github/ubuntu/cpu/newest_stable/continuous.cfg

@@ -1,9 +1,5 @@
 build_file: "gnn/kokoro/github/ubuntu/cpu/build_versioned.sh"
 
-env_vars: {
-  key: "USE_BAZEL_VERSION"
-  value: "7.4.1"  # TODO - b/390391579: Unpin once bazel 8 works.
-}
 env_vars: {
   key: "PYTHON_VERSION"
   value: "3.11"

kokoro/github/ubuntu/cpu/newest_stable/presubmit.cfg

@@ -1,9 +1,5 @@
 build_file: "gnn/kokoro/github/ubuntu/cpu/build_versioned.sh"
 
-env_vars: {
-  key: "USE_BAZEL_VERSION"
-  value: "7.4.1"  # TODO - b/390391579: Unpin once bazel 8 works.
-}
 env_vars: {
   key: "PYTHON_VERSION"
   value: "3.11"

kokoro/github/ubuntu/cpu/oldest/continuous.cfg

@@ -1,9 +1,5 @@
 build_file: "gnn/kokoro/github/ubuntu/cpu/build_versioned.sh"
 
-env_vars: {
-  key: "USE_BAZEL_VERSION"
-  value: "7.4.1"  # TODO - b/390391579: Unpin once bazel 8 works.
-}
 env_vars: {
   key: "PYTHON_VERSION"
   value: "3.11"

kokoro/github/ubuntu/cpu/oldest/presubmit.cfg

@@ -1,9 +1,5 @@
 build_file: "gnn/kokoro/github/ubuntu/cpu/build_versioned.sh"
 
-env_vars: {
-  key: "USE_BAZEL_VERSION"
-  value: "7.4.1"  # TODO - b/390391579: Unpin once bazel 8 works.
-}
 env_vars: {
   key: "PYTHON_VERSION"
   value: "3.11"

kokoro/github/ubuntu/cpu/presubmit.cfg

@@ -1,9 +1,5 @@
 build_file: "gnn/kokoro/github/ubuntu/cpu/build.sh"
 
-env_vars: {
-  key: "USE_BAZEL_VERSION"
-  value: "7.4.1"  # TODO - b/390391579: Unpin once bazel 8 works.
-}
 env_vars: {
   key: "PYTHON_VERSION"
   value: "3.11"