Attempt to fix label test.

explosion · Oct 30, 2023 · 95a89e9 · 95a89e9
1 parent e8cb182
commit 95a89e9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/spacy_llm/tests/tasks/test_ner.py b/spacy_llm/tests/tasks/test_ner.py
@@ -994,7 +994,8 @@ def test_add_label():
 
     for label, definition in [
         ("PERSON", "Every person with the name Jack"),
-        ("LOCATION", None),
+        ("LOCATION", "A geographical location, like a country or a city"),
+        ("COMPANY", None),
     ]:
         llm.add_label(label, definition)
     doc = nlp(text)