Use env vars for benching host

capablevms · Nov 1, 2024 · cbbd6d1 · cbbd6d1
1 parent 1956ae7
commit cbbd6d1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/scripts/do_benchmarks.py b/scripts/do_benchmarks.py
@@ -4,15 +4,17 @@
 import os
 import shlex
 import subprocess
+import sys
 
 from fabric import Connection
 
 ################################################################################
 # Constants
 ################################################################################
 
-BENCH_HOST = "10.81.18.119"
-BENCH_USER = "al2510"
+guard_getenv = lambda envname: os.getenv(envname) or sys.exit(f"Missing env var `{envname}`!")
+BENCH_HOST = guard_getenv("COMP_BENCH_HOST")
+BENCH_USER = guard_getenv("COMP_BENCH_USER")
 
 BENCH_RUN_PATH = "/home/0152la/bench-script"
 BENCH_RUN_LIBS_PATH = f"{BENCH_RUN_PATH}/libs"