responded to comment

mideind · Oct 25, 2023 · 6f13832 · 6f13832
1 parent de9626b
commit 6f13832
Showing 1 changed file with 1 addition and 9 deletions.
diff --git 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: