Resolved issue with environment variables (#35)

Fix some issues with env variables. --------- Signed-off-by: Ghazi Sarwat Syed <[email protected]>
fmperf-project · Nov 4, 2024 · f40afc5 · f40afc5
1 parent 7cb317b
commit f40afc5
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/fmperf/Cluster.py b/fmperf/Cluster.py
@@ -248,7 +248,7 @@ def generate_workload(
                             {
                                 "name": "fmaas-perf",
                                 "image": workload.image,
-                                "imagePullPolicy": "IfNotPresent",
+                                "imagePullPolicy": "Always",
                                 "env": workload.get_env(target, model, outfile),
                                 "command": ["/bin/bash", "-ce"],
                                 "args": workload.get_args(),
@@ -402,7 +402,7 @@ def evaluate(
                         "containers": [
                             {
                                 "name": "fmaas-perf",
-                                "imagePullPolicy": "IfNotPresent",
+                                "imagePullPolicy": "Always",
                                 "image": workload.spec.image,
                                 "env": env,
                                 "command": ["/bin/bash", "-ce"],

diff --git a/fmperf/loadgen/run.py b/fmperf/loadgen/run.py
@@ -12,12 +12,12 @@
 from fmperf.utils import parse_results
 from datetime import datetime
 from .collect_energy import collect_metrics, summarize_energy
-from fmperf.utils.constants import REQUESTS_DIR, REQUESTS_FILENAME, RESULTS_ALL_FILENAME
+from fmperf.utils.constants import REQUESTS_DIR, REQUESTS_FILENAME, RESULTS_FILENAME
 
 
 def run(result_filename=None):
     if result_filename is None:
-        result_filename = RESULTS_ALL_FILENAME
+        result_filename = RESULTS_FILENAME
 
     def get_streaming_response_tgis(response):
         stop = False

diff --git a/fmperf/utils/constants.py b/fmperf/utils/constants.py
@@ -3,3 +3,4 @@
 REQUESTS_DIR = os.environ.get("REQUESTS_DIR", ".")
 REQUESTS_FILENAME = os.environ["REQUESTS_FILENAME"]
 RESULTS_ALL_FILENAME = os.environ.get("RESULTS_ALL_FILENAME", None)
+RESULTS_FILENAME = os.environ.get("RESULTS_FILENAME", None)