From 45e03ad0c0be0e1722c6e6cf729d87196101a133 Mon Sep 17 00:00:00 2001
From: Adil Hafeez <adil@katanemo.com>
Date: Wed, 30 Oct 2024 12:01:24 -0700
Subject: [PATCH] better prints

---
 arch/tools/cli/main.py     | 2 --
 e2e_tests/run_e2e_tests.sh | 6 ++++++
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/arch/tools/cli/main.py b/arch/tools/cli/main.py
index 183d23d2..a5b5c07e 100644
--- a/arch/tools/cli/main.py
+++ b/arch/tools/cli/main.py
@@ -183,8 +183,6 @@ def up(file, path, service):
 
     # remove duplicates
     access_keys = set(access_keys)
-    # remove the $ from the access_keys
-    access_keys = [item[1:] if item.startswith("$") else item for item in access_keys]
 
     if access_keys:
         if file:
diff --git a/e2e_tests/run_e2e_tests.sh b/e2e_tests/run_e2e_tests.sh
index ff8eeaa1..eb9f82fe 100644
--- a/e2e_tests/run_e2e_tests.sh
+++ b/e2e_tests/run_e2e_tests.sh
@@ -26,10 +26,14 @@ cd ../demos/function_calling
 docker compose up api_server --build -d
 cd -
 
+print_disk_usage
+
 log building model server
 log =====================
 cd ../model_server
 poetry install 2>&1 >> ../build.log
+print_disk_usage
+
 log starting model server
 log =====================
 mkdir -p ~/archgw_logs
@@ -42,7 +46,9 @@ cd -
 log building llm and prompt gateway rust modules
 log ============================================
 cd ../arch
+print_disk_usage
 docker build  -f Dockerfile .. -t katanemo/archgw -q
+print_disk_usage
 log starting the arch gateway service
 log =================================
 docker compose -f docker-compose.e2e.yaml down