diff --git a/src/reynir_correct/pattern.py b/src/reynir_correct/pattern.py index 7dfe23b..7842fc4 100644 --- a/src/reynir_correct/pattern.py +++ b/src/reynir_correct/pattern.py @@ -226,15 +226,7 @@ def get_wordform(self, lemma, cat, variants): given a set of variants""" # Get rid of argument variants in verbs: - variants = list([x for x in variants if not x.isdigit()]) - realvars = [] - for x in variants: - if x.isdigit(): - continue - if x in SKIPVARS: - continue - else: - realvars.append(x) + realvars = [x for x in variants if not x.isdigit() and x not in SKIPVARS] wordforms = BIN.lookup_variants(lemma, cat, realvars) if not wordforms: