Spacy integration for the German IWNLP lemmatizer

iwnlp.py

from spacy import Language
from spacy.lang.de import German
from spacy.pipeline.lemmatizer import *
from spacy.tokens import Doc
from iwnlp.iwnlp_wrapper import IWNLPWrapper

class IWNLPLemmatizer:
    def __init__(self):
        self.lemmatizer = IWNLPWrapper(lemmatizer_path='./zata/language_data/IWNLP.Lemmatizer_20181001.json')

    def __call__(self, doc: Doc) -> Doc:
        for token in doc:
            lemmas = self.lemmatizer.lemmatize(token.text, pos_universal_google=token.pos_)
            if lemmas != None and len(lemmas) > 0:
                token.lemma_ = lemmas[0]
        return doc

@German.factory("iwnlp_lemmatizer")
def create_en_normalizer(nlp, name):
    return IWNLPLemmatizer()