From 582b26587ff543d41e615478cd3b0ef1cf149028 Mon Sep 17 00:00:00 2001
From: Alistair Muldal <alimuldal@google.com>
Date: Tue, 13 Apr 2021 17:16:26 +0100
Subject: [PATCH] Enable Travis tests for `option_keyboard`

PiperOrigin-RevId: 368228512
---
 .travis.yml            | 1 +
 option_keyboard/run.sh | 5 +++--
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/.travis.yml b/.travis.yml
index 1ddb877..1aa0850 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -15,6 +15,7 @@ env:
     - PROJECT="kfac_ferminet_alpha"
     - PROJECT="learning_to_simulate"
     - PROJECT="noise_covariance"
+    - PROJECT="option_keyboard"
     # - PROJECT="polygen"  # TODO(b/184847339): Fix and re-enable
     - PROJECT="transporter"
     - PROJECT="tvt"
diff --git a/option_keyboard/run.sh b/option_keyboard/run.sh
index f27d231..58d2802 100755
--- a/option_keyboard/run.sh
+++ b/option_keyboard/run.sh
@@ -13,8 +13,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-python3 -m venv ok_venv
-source ok_venv/bin/activate
+python3 -m venv /tmp/ok_venv
+source /tmp/ok_venv/bin/activate
+pip install -U pip
 pip install -r option_keyboard/requirements.txt
 
 python3 -m option_keyboard.run_dqn_test