Fix edge cases of empty inputs for MTL

hankcs · hankcs · commit 15bb02f3d503 · 2022-04-15T12:49:09.000-04:00
(cherry picked from commit f123d36) (cherry picked from commit 1fa9cac)
diff --git a/hanlp/components/mtl/multi_task_learning.py b/hanlp/components/mtl/multi_task_learning.py
@@ -475,7 +475,7 @@ def predict(self,
             A :class:`~hanlp_common.document.Document`.
         """
         doc = Document()
-        if not data:
+        if not data or not any(data):
             return doc
 
         target_tasks = resolved_tasks or self.resolve_tasks(tasks, skip_tasks)
diff --git a/hanlp/version.py b/hanlp/version.py
@@ -2,7 +2,7 @@
 # Author: hankcs
 # Date: 2019-12-28 19:26
 
-__version__ = '2.1.0-beta.22'
+__version__ = '2.1.0-beta.23'
 """HanLP version"""
 
 
diff --git a/tests/test_mtl.py b/tests/test_mtl.py
@@ -29,6 +29,7 @@ def test_mtl_empty_str(self):
         mtl([' '])
         mtl(['', ' '])
         mtl(['', ' ', 'good'])
+        mtl([[]], skip_tasks='tok*')
 
     def test_skip_tok(self):
         pre_tokenized_sents = [
