Internal change.

PiperOrigin-RevId: 320226296
Change-Id: I36c938ee6ac976379bc4d391db7f7e1634a677fe

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
 

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