diff --git a/tensorflow/tools/ci_build/builds/pip_new.sh b/tensorflow/tools/ci_build/builds/pip_new.sh index 72f1b58208772ca8070829ad94bb8950a737da3d..e3f375edb9b58b6446cee3145aa855079236f3b1 100755 --- a/tensorflow/tools/ci_build/builds/pip_new.sh +++ b/tensorflow/tools/ci_build/builds/pip_new.sh @@ -458,6 +458,12 @@ install_tensorflow_pip() { # ImportError: No module named builtins ${PIP_BIN_PATH} install --upgrade "future>=0.17.1" || \ die "Error: future install, upgrade FAILED" + + # Install the gast package in the virtualenv. Installing it in user system + # packages does not appear to port it over when creating a virtualenv. + ${PIP_BIN_PATH} install --upgrade "gast==0.2.2" || \ + die "Error: gast install, upgrade FAILED" + } run_test_with_bazel() {