Source code for OpenAttack.metric.selectors.fluency

from .base import MetricSelector
class Fluency(MetricSelector):
    """
    :English: :py:class:`.GPT2LM`
    :Chinese: :py:class:`.GPT2LMChinese`
    """

    def _select(self, lang):
        if lang.name == "english":
            from ..algorithms.gptlm import GPT2LM
            return GPT2LM()
        if lang.name == "chinese":
            from ..algorithms.gptlm import GPT2LMChinese
            return GPT2LMChinese()