Return tf ner grouped test to original

huggingface · Jul 4, 2020 · 4a7a483 · 4a7a483
1 parent 7de9685
commit 4a7a483
Showing 1 changed file with 2 additions and 12 deletions.
diff --git a/tests/test_pipelines.py b/tests/test_pipelines.py
@@ -200,19 +200,9 @@ def test_tf_ner(self):
     @require_tf
     def test_tf_ner_grouped(self):
         mandatory_keys = {"entity_group", "word", "score"}
-        valid_inputs = [
-            "Consuelo Araújo Noguera, ministra de cultura del presidente Andrés Pastrana (1998.2002) fue asesinada por las Farc luego de haber permanecido secuestrada por algunos meses."
-        ]
-        expected_check_keys = ["entity_group", "word"]
-        for model_name in GROUPED_NER_FINETUNED_MODELS:
+        for model_name in NER_FINETUNED_MODELS:
             nlp = pipeline(task="ner", model=model_name, tokenizer=model_name, framework="tf", grouped_entities=True)
-            self._test_mono_column_pipeline(
-                nlp,
-                valid_inputs,
-                mandatory_keys,
-                expected_multi_result=expected_grouped_ner_result,
-                expected_check_keys=expected_check_keys,
-            )
+            self._test_mono_column_pipeline(nlp, VALID_INPUTS, mandatory_keys)
 
     @require_torch
     def test_torch_sentiment_analysis(self):