From 6f13832fd09c06e38b336d4d66df3531e839cfbc Mon Sep 17 00:00:00 2001
From: thorunna <thar@hi.is>
Date: Wed, 25 Oct 2023 09:45:58 +0000
Subject: [PATCH] responded to comment

---
 src/reynir_correct/pattern.py | 10 +---------
 1 file changed, 1 insertion(+), 9 deletions(-)

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: