Shaltiel commited on
Commit
51bfeec
1 Parent(s): d876fb6

Added exception printing

Browse files
src/backend/run_eval_suite_lighteval.py CHANGED
@@ -50,7 +50,8 @@ def run_evaluation(eval_request: EvalRequest, task_names: str, batch_size: int,
50
 
51
  dumped = json.dumps(results, indent=2)
52
  print(dumped)
53
- except Exception: # if eval failed, we force a cleanup
 
54
  env_config = EnvConfig(token=TOKEN, cache_dir=args.cache_dir)
55
 
56
  model_config = create_model_config(args=args, accelerator=accelerator)
 
50
 
51
  dumped = json.dumps(results, indent=2)
52
  print(dumped)
53
+ except Exception as ex: # if eval failed, we force a cleanup
54
+ print(ex)
55
  env_config = EnvConfig(token=TOKEN, cache_dir=args.cache_dir)
56
 
57
  model_config = create_model_config(args=args, accelerator=accelerator)