From 95a89e91cdbf5a5713a753c681400d540a5f373b Mon Sep 17 00:00:00 2001 From: Raphael Mitsch Date: Mon, 30 Oct 2023 15:31:24 +0100 Subject: [PATCH] Attempt to fix label test. --- spacy_llm/tests/tasks/test_ner.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/spacy_llm/tests/tasks/test_ner.py b/spacy_llm/tests/tasks/test_ner.py index 97512ca6..adb8c436 100644 --- 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)