diff --git a/src/lighteval/tasks/extended/lcb/codegen_metrics.py b/src/lighteval/tasks/extended/lcb/codegen_metrics.py index 98fad8858..8e87bb99b 100644 --- a/src/lighteval/tasks/extended/lcb/codegen_metrics.py +++ b/src/lighteval/tasks/extended/lcb/codegen_metrics.py @@ -53,7 +53,10 @@ from tqdm import tqdm -sys.set_int_max_str_digits(50000) +try: + sys.set_int_max_str_digits(50000) +except AttributeError: + pass os.environ["TOKENIZERS_PARALLELISM"] = "false"