In TF-GNN's Kokoro scripts, do a final pip list for clarity.

PiperOrigin-RevId: 722646358
tensorflow · Feb 3, 2025 · 83e110b · 83e110b
1 parent 5ed86a1
commit 83e110b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/kokoro/github/ubuntu/cpu/build.sh b/kokoro/github/ubuntu/cpu/build.sh
@@ -51,6 +51,10 @@ perl  -i -lpe '$k+= s/tensorflow>=2\.[0-9]+\.[0-9]+(,<=?[0-9.]+)?;/tf-nightly;/g
 python3 setup.py bdist_wheel
 pip uninstall -y tensorflow_gnn
 pip install dist/tensorflow_gnn-*.whl
+
+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=

diff --git a/kokoro/github/ubuntu/cpu/build_versioned.sh b/kokoro/github/ubuntu/cpu/build_versioned.sh
@@ -63,4 +63,7 @@ python3 setup.py bdist_wheel
 pip uninstall -y tensorflow_gnn
 pip install dist/tensorflow_gnn-*.whl
 
+echo "Final packages after all pip commands:"
+pip list
+
 bazel test --test_env=TF_USE_LEGACY_KERAS --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/...