catch all exceptions in the benchmark

This commit is contained in:
Paul Gauthier 2024-05-04 17:52:46 -07:00
parent 3e4fca2675
commit 9cdd9e12c3

View file

@ -7,6 +7,7 @@ import re
import shutil import shutil
import subprocess import subprocess
import time import time
import traceback
from collections import defaultdict from collections import defaultdict
from json.decoder import JSONDecodeError from json.decoder import JSONDecodeError
from pathlib import Path from pathlib import Path
@ -927,7 +928,17 @@ def get_replayed_content(replay_dname, test_dname):
return "".join(res) return "".join(res)
def run_test( def run_test(*args, **kwargs):
try:
return run_test_real(*args, **kwargs)
except Exception as err:
print("=" * 40)
print("Test failed")
print(err)
traceback.print_exc()
def run_test_real(
original_dname, original_dname,
testdir, testdir,
model_name, model_name,