Update .buildkite/lm-eval-harness/test_lm_eval_correctness.py

Co-authored-by: Cyrus Leung <[email protected]>
vllm-project · Aug 6, 2024 · bf0c2f2 · bf0c2f2
1 parent 19a356b
commit bf0c2f2
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/.buildkite/lm-eval-harness/test_lm_eval_correctness.py b/.buildkite/lm-eval-harness/test_lm_eval_correctness.py
@@ -23,10 +23,7 @@
 
 
 def launch_lm_eval(eval_config):
-    try:
-        trust_remote_code = eval_config['trust_remote_code']
-    except Exception:
-        trust_remote_code = False
+    trust_remote_code = eval_config.get('trust_remote_code', False)
 
     model_args = f"pretrained={eval_config['model_name']}," \
                  f"tensor_parallel_size={TP_SIZE}," \