From f0a3ce5a45afa0776ff5bd83c448a556024a01b3 Mon Sep 17 00:00:00 2001 From: hardianlawi Date: Mon, 10 Jan 2022 15:48:14 +0000 Subject: [PATCH] Convert lambda to regular function --- src/benepar/integrations/spacy_extensions.py | 25 ++++++++++++-------- 1 file changed, 15 insertions(+), 10 deletions(-) diff --git a/src/benepar/integrations/spacy_extensions.py b/src/benepar/integrations/spacy_extensions.py index 572dc45..c9b4cbd 100644 --- a/src/benepar/integrations/spacy_extensions.py +++ b/src/benepar/integrations/spacy_extensions.py @@ -161,16 +161,21 @@ def install_spacy_extensions(): Span.set_extension("parent", getter=get_parent_span) Span.set_extension("children", getter=get_child_spans) - Token.set_extension( - "labels", getter=lambda token: get_labels(token.doc[token.i : token.i + 1]) - ) - Token.set_extension( - "parse_string", - getter=lambda token: parse_string(token.doc[token.i : token.i + 1]), - ) - Token.set_extension( - "parent", getter=lambda token: get_parent_span(token.doc[token.i : token.i + 1]) - ) + Token.set_extension("labels", getter=_labels) + Token.set_extension("parse_string", getter=_parse_string) + Token.set_extension("parent", getter=_get_parent_span) + + +def _labels(token): + return get_labels(token.doc[token.i : token.i + 1]) + + +def _parse_string(token): + return parse_string(token.doc[token.i : token.i + 1]) + + +def _get_parent_span(token): + return get_parent_span(token.doc[token.i : token.i + 1]) try: