ref: replace if-else chain with dict lookup for response computation

meta-llama · Apr 23, 2024 · 4fc17ce · 4fc17ce
1 parent 9b864b8
commit 4fc17ce
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/CybersecurityBenchmarks/benchmark/run.py b/CybersecurityBenchmarks/benchmark/run.py
@@ -164,11 +164,11 @@ def main(
 
     materialized_llms_under_test = []
     if not args.use_precomputed_responses:
-        materialized_llms_under_test = llms_under_test or [
-            default_llm_creator(specification) for specification in args.llm_under_test
-        ]
-        if len(materialized_llms_under_test) == 0:
+        if not args.llm_under_test:
             raise ValueError("No LLMs were specified.")
+        materialized_llms_under_test = llms_under_test or [  
+            default_llm_creator(specification) for specification in args.llm_under_test  
+        ]
 
     stat_response_path = Path(args.stat_path) if args.stat_path else None
     judge_response_path = (