From fd7cc7607fc87c793dbd78d50e96216d2411d1a3 Mon Sep 17 00:00:00 2001
From: Michael Reneer <michaelreneer@google.com>
Date: Tue, 10 Oct 2023 12:50:18 -0700
Subject: [PATCH] Update the Python package scripts to use Python 3.10
 explicitly.

Failure:

1. From the build scripts:

```
+ ./tools/build_pip_package.sh
Python 3.8.10
```

2. But `tensorflow-probability` does not support Python 3.8, https://pypi.org/project/tensorflow-probability/

The fix is to update TFP build scripts to use Python 3.10 explicitly because it appears there are unsupported version of Python on the build machines.

PiperOrigin-RevId: 572339377
---
 tools/build_pip_package.sh   | 2 +-
 tools/publish_pip_package.sh | 2 +-
 tools/test_pip_package.sh    | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/tools/build_pip_package.sh b/tools/build_pip_package.sh
index de90d24..529e6f0 100755
--- a/tools/build_pip_package.sh
+++ b/tools/build_pip_package.sh
@@ -22,7 +22,7 @@ main() {
   trap "rm -rf ${temp_dir}" EXIT
 
   # Create a virtual environment
-  python3 -m venv "${temp_dir}/venv"
+  python3.10 -m venv "${temp_dir}/venv"
   source "${temp_dir}/venv/bin/activate"
   python --version
   pip install --upgrade pip
diff --git a/tools/publish_pip_package.sh b/tools/publish_pip_package.sh
index 3faa3a1..d29b32b 100755
--- a/tools/publish_pip_package.sh
+++ b/tools/publish_pip_package.sh
@@ -22,7 +22,7 @@ main() {
   trap "rm -rf ${temp_dir}" EXIT
 
   # Create a virtual environment
-  python3 -m venv "${temp_dir}/venv"
+  python3.10 -m venv "${temp_dir}/venv"
   source "${temp_dir}/venv/bin/activate"
   python --version
   pip install --upgrade pip
diff --git a/tools/test_pip_package.sh b/tools/test_pip_package.sh
index afdc4b7..0c582a5 100755
--- a/tools/test_pip_package.sh
+++ b/tools/test_pip_package.sh
@@ -22,7 +22,7 @@ main() {
   trap "rm -rf ${temp_dir}" EXIT
 
   # Create a virtual environment
-  python3 -m venv "${temp_dir}/venv"
+  python3.10 -m venv "${temp_dir}/venv"
   source "${temp_dir}/venv/bin/activate"
   python --version
   pip install --upgrade pip