From a170ecd5ce41777982aefc0ff83c82d793704cbc Mon Sep 17 00:00:00 2001
From: Eugene Brevdo <ebrevdo@google.com>
Date: Wed, 8 Jul 2020 11:15:20 -0700
Subject: [PATCH] Internal change.

PiperOrigin-RevId: 320226296
Change-Id: I36c938ee6ac976379bc4d391db7f7e1634a677fe
---
 tests_nightly_pypi.sh | 7 -------
 tests_release.sh      | 6 +++++-
 2 files changed, 5 insertions(+), 8 deletions(-)

diff --git a/tests_nightly_pypi.sh b/tests_nightly_pypi.sh
index 8423571e5..be1779262 100755
--- a/tests_nightly_pypi.sh
+++ b/tests_nightly_pypi.sh
@@ -40,11 +40,6 @@ run_tests() {
     # Install pypi package
     pip install tf-agents-nightly
 
-  elif [[ $2 == "preview" ]] ; then
-    pip install tf-nightly-2.0-preview
-
-    # Install pypi package
-    pip install tf-agents-nightly
   else
     echo "Error unknown option only [test|official]"
     exit
@@ -58,8 +53,6 @@ run_tests() {
   deactivate
 }
 
-# Test on Python2.7
-run_tests "python2.7" $1
 # Test on Python3.6
 run_tests "python3.6" $1
 
diff --git a/tests_release.sh b/tests_release.sh
index 26c3972e0..09714f605 100755
--- a/tests_release.sh
+++ b/tests_release.sh
@@ -66,14 +66,18 @@ run_tests() {
   TMP=$(mktemp -d)
   # Creates and activates a virtualenv to run the build and unittests in.
   VENV_PATH=${TMP}/virtualenv/$1
-  virtualenv "${VENV_PATH}"
+  virtualenv -p ~/.pyenv/versions/${PYTHON_VERSION}/bin/python "${VENV_PATH}"
   source ${VENV_PATH}/bin/activate
 
+  # Print the version of python
+  python --version
+  which pip
 
   # TensorFlow is not set as a dependency of TF-Agents because there are many
   # different TensorFlow versions a user might want and installed.
   if [ "$RELEASE_TYPE" = "nightly" ]; then
     pip install tf-nightly
+    pip install dm-reverb-nightly
 
     # Run the tests
     python setup.py test