From 5527e0f3285d45c338612e3938f023d643722abb Mon Sep 17 00:00:00 2001
From: Lidi Zheng <lidiz@google.com>
Date: Tue, 29 Sep 2020 13:28:45 -0700
Subject: [PATCH] Simplify the check with existing if

---
 tools/run_tests/helper_scripts/build_python.sh | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/tools/run_tests/helper_scripts/build_python.sh b/tools/run_tests/helper_scripts/build_python.sh
index aba06dddc84..331f10e838a 100755
--- a/tools/run_tests/helper_scripts/build_python.sh
+++ b/tools/run_tests/helper_scripts/build_python.sh
@@ -188,15 +188,11 @@ esac
 pip_install --upgrade setuptools==44.1.1
 pip_install --upgrade pip==19.3.1
 pip_install --upgrade cython
-if [ -z $($VENV_PYTHON -V | grep "Python 2.7") ]; then
-  pip_install --upgrade six enum34 protobuf
-else
-  pip_install --upgrade six protobuf
-fi
+pip_install --upgrade six protobuf
 
 if [ "$("$VENV_PYTHON" -c "import sys; print(sys.version_info[0])")" == "2" ]
 then
-  pip_install futures
+  pip_install --upgrade futures enum34
 fi
 
 pip_install_dir "$ROOT"