From f95237012ffe3c1eeab4f616a29035a4702278cd Mon Sep 17 00:00:00 2001 From: Adriane Boyd Date: Wed, 14 Dec 2022 09:28:14 +0100 Subject: [PATCH] Rename keys instead of using a custom getter in test --- spacy/tests/test_language.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/spacy/tests/test_language.py b/spacy/tests/test_language.py index 2977f5352..03790eb86 100644 --- a/spacy/tests/test_language.py +++ b/spacy/tests/test_language.py @@ -186,16 +186,13 @@ def test_evaluate_multiple_textcat_separate(en_vocab): with custom scorers.""" def custom_textcat_score(examples, **kwargs): - def custom_getter(doc, _): - return doc.cats - - return Scorer.score_cats( + scores = Scorer.score_cats( examples, - "custom_cats", + "cats", multi_label=False, - getter=custom_getter, **kwargs, ) + return {f"custom_{k}": v for k, v in scores.items()} @spacy.registry.scorers("test_custom_textcat_scorer") def make_custom_textcat_scorer():