refactor code to avoid flake8 warning

annif/analyzer/estnltk.py

@@ -2,12 +2,19 @@
 
 from __future__ import annotations
 
+import importlib
+
 from . import analyzer
 
 
 class EstNLTKAnalyzer(analyzer.Analyzer):
     name = "estnltk"
 
+    @staticmethod
+    def is_available() -> bool:
+        # return True iff EstNLTK is installed
+        return importlib.util.find_spec("estnltk") is not None
+
     def __init__(self, param: str, **kwargs) -> None:
         self.param = param
         super().__init__(**kwargs)
@@ -17,9 +24,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

tests/test_analyzer_estnltk.py

@@ -3,8 +3,12 @@
 import pytest
 
 import annif.analyzer
+import annif.analyzer.estnltk
 
-estnltk = pytest.importorskip("estnltk")
+pytestmark = pytest.mark.skipif(
+    not annif.analyzer.estnltk.EstNLTKAnalyzer.is_available(),
+    reason="EstNLTK is required",
+)
 
 
 def test_estnltk_tokenize_words():