Pass trust_remote_code to fix test_token_classification_parity

huggingface · Sep 12, 2024 · e9e277a · e9e277a
1 parent c476380
commit e9e277a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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}",