refactor code to avoid flake8 warning

NatLibFi · Nov 12, 2024 · d2a0051 · d2a0051
1 parent a076607
commit d2a0051
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/annif/analyzer/estnltk.py b/annif/analyzer/estnltk.py
@@ -17,9 +17,8 @@ def tokenize_words(self, text: str, filter: bool = True) -> list[str]:
 
         txt = estnltk.Text(text.strip())
         txt.tag_layer()
-        lemmas = [
+        return [
             lemma
-            for lemma in [l[0] for l in txt.lemma]
+            for lemma in [lemmas[0] for lemmas in txt.lemma]
             if (not filter or self.is_valid_token(lemma))
         ]
-        return lemmas