From e9e277a98020e31a4ca0f026b47f2a354081e6f0 Mon Sep 17 00:00:00 2001 From: Albert Villanova del Moral <8515462+albertvillanova@users.noreply.github.com> Date: Thu, 12 Sep 2024 11:40:49 +0200 Subject: [PATCH] Pass trust_remote_code to fix test_token_classification_parity --- tests/test_trainer_evaluator_parity.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_trainer_evaluator_parity.py b/tests/test_trainer_evaluator_parity.py index b513da886..36c5ccd55 100644 --- a/tests/test_trainer_evaluator_parity.py +++ b/tests/test_trainer_evaluator_parity.py @@ -283,6 +283,7 @@ def test_token_classification_parity(self): f"python examples/pytorch/token-classification/run_ner.py" f" --model_name_or_path {model_name}" f" --dataset_name conll2003" + f" --trust_remote_code" f" --do_eval" f" --output_dir {os.path.join(self.dir_path, 'tokenclassification_conll2003_transformers')}" f" --max_eval_samples {n_samples}",