languages: fix predictions for editorial/introduction

CoLRev-Environment · Mar 28, 2023 · 93fc7e8 · 93fc7e8
1 parent 062a415
commit 93fc7e8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/colrev/env/language_service.py b/colrev/env/language_service.py
@@ -34,6 +34,9 @@ def __init__(self) -> None:
     def compute_language_confidence_values(self, *, text: str) -> list:
         """Computes the most likely languages of a string and their language codes"""
 
+        if text.lower() in ["editorial", "introduction"]:
+            return [("eng", 1.0)]
+
         predictions = (
             self.__lingua_language_detector.compute_language_confidence_values(
                 text=text

diff --git a/tests/1_env/language_service_test.py b/tests/1_env/language_service_test.py
@@ -50,6 +50,14 @@ def language_service() -> colrev.env.language_service.LanguageService:  # type:
             "An Integrated Framework for Understanding Digital Work in Organizations",
             ("eng", 0.9),
         ),
+        (
+            "Editorial",
+            ("eng", 0.9),
+        ),
+        (
+            "Introduction",
+            ("eng", 0.9),
+        ),
     ],
 )
 def test_compute_language_confidence_values(