mirrors/LibreTranslate
1
1
Fork 1
mirror of https://github.com/LibreTranslate/LibreTranslate.git synced 2024-06-10 17:39:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix errors in detect.py

Browse source
This commit is contained in:
Joshix-1 2023-01-18 19:00:00 +00:00
parent 0f70f9a4c4
commit ea32b1908b
1 changed files with 6 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
libretranslate/detect.py
View file

@ -44,17 +44,15 @@ class BaseDetector(ABC):
def __init__(
self,
text: str,
allowed_languages: "list[str]",
quiet: bool = False,
allowed_languages: "Iterable[str] | None" = None,
) -> None:
"""Detector of the language used in `text`.
Args:
text (string): unicode string.
"""
self.allowed_languages: "frozenset[str]" = frozenset(
self.supported_languages()
if allowed_languages is None
else [string.upper() for string in allowed_languages]
[string.upper() for string in allowed_languages]
)
# self.__text = text
self.reliable: bool = True
@ -127,7 +125,7 @@ class LinguaDetector(BaseDetector):
def supported_languages(cls) -> "list[str]":
"""Returns a list of the languages that can be detected by pycld2."""
return [
lang.iso_code_639_1.name
lang.name.capitalize()
for lang in lingua.Language.all()
]
@ -188,10 +186,10 @@ class Detector(BaseDetector):
def test_Detector():
for lang in Detector.supported_languages():
assert len(lang) == 2 and lang.upper() == lang, lang
assert Detector("Neuland").language.code == "DE"
assert lang.capitalize() == lang, lang
assert Detector("Neuland", allowed_languages=["de", "en", "it", "fr"]).language.code == "DE"
# https://github.com/LibreTranslate/LibreTranslate/issues/247
assert Detector("Tout philosophe a deux philosophies : la sienne et celle de Spinoza.").language.code == "FR"
assert Detector("Tout philosophe a deux philosophies : la sienne et celle de Spinoza.", allowed_languages=["de", "en", "it", "fr"]).language.code == "FR"
if __name__ == "__main__":