Source code for OpenAttack.metric.selectors.grammar

from .base import MetricSelector
class GrammaticalErrors(MetricSelector):
    """
    :English: :py:class:`.LanguageTool`
    :Chinese: :py:class:`.LanguageToolChinese`
    """

    def _select(self, lang):
        if lang.name == "english":
            from ..algorithms.language_tool import LanguageTool
            return LanguageTool()
        if lang.name == "chinese":
            from ..algorithms.language_tool import LanguageToolChinese
            return LanguageToolChinese()