GH-2534: fix typing in regexp_tagger.py

flairNLP · Dec 6, 2021 · 7fdf215 · 7fdf215
1 parent d4cf148
commit 7fdf215
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/flair/models/regexp_tagger.py b/flair/models/regexp_tagger.py
@@ -95,6 +95,8 @@ def predict(self, sentences: Union[List[Sentence], Sentence]) -> List[Sentence]:
         """
             Predict the given sentences according to the registered mappings.
         """
+        if not isinstance(sentences, list):
+            sentences = [sentences]
         if not sentences:
             return sentences