From 276a04238ea87f4933cc30949931b00fba32f7ee Mon Sep 17 00:00:00 2001
From: Bernard Han <103850980+bernardhan33@users.noreply.github.com>
Date: Wed, 10 Jul 2024 13:35:40 -0700
Subject: [PATCH] Fix continuous and presubmit tests (#47)

* attempt to fix continuous test by adding apt-get update

* add '-y'

* add more '-y'
---
 kokoro/build.sh     | 4 +++-
 kokoro/presubmit.sh | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/kokoro/build.sh b/kokoro/build.sh
index d48802d5..c6140eca 100644
--- a/kokoro/build.sh
+++ b/kokoro/build.sh
@@ -21,8 +21,10 @@ set -e
 cd "${KOKORO_ARTIFACTS_DIR}/github/dataflux-pytorch"
 
 function setup_virtual_envs() {
+    sudo apt-get -y update
+
     echo Setting up Python virtual environment.
-    sudo apt install python3.8-venv
+    sudo apt install -y python3.8-venv
     python3 -m venv venv
     source venv/bin/activate
 }
diff --git a/kokoro/presubmit.sh b/kokoro/presubmit.sh
index b5241cef..0c5b2dbf 100644
--- a/kokoro/presubmit.sh
+++ b/kokoro/presubmit.sh
@@ -21,8 +21,10 @@ set -e
 cd "${KOKORO_ARTIFACTS_DIR}/github/dataflux-pytorch"
 
 function setup_virtual_envs() {
+    sudo apt-get -y update
+
     echo Setting up Python virtual environment.
-    sudo apt install python3.8-venv
+    sudo apt install -y python3.8-venv
     python3 -m venv venv
     source venv/bin/activate
 }