Source code for data_juicer.ops.mapper.fix_unicode_mapper

from data_juicer.utils.lazy_loader import LazyLoader

from ..base_op import OPERATORS, Mapper

ftfy = LazyLoader('ftfy', 'ftfy')

OP_NAME = 'fix_unicode_mapper'


[docs] @OPERATORS.register_module(OP_NAME) class FixUnicodeMapper(Mapper): """Mapper to fix unicode errors in text samples.""" _batched_op = True
[docs] def __init__(self, normalization: str = None, *args, **kwargs): """ Initialization method. :param normalization: the specified form of Unicode normalization mode, which can be one of ['NFC', 'NFKC', 'NFD', and 'NFKD'], default 'NFC'. :param args: extra args :param kwargs: extra args """ super().__init__(*args, **kwargs) if normalization and len(normalization) > 0: self.normalization = normalization.upper() else: self.normalization = 'NFC' if self.normalization.upper() not in ['NFC', 'NFKC', 'NFD', 'NFKD']: raise ValueError(f'Normalization mode [{normalization}] is not ' 'supported. Can only be one of ' '["NFC", "NFKC", "NFD", "NFKD"]')
[docs] def process_batched(self, samples): samples[self.text_key] = [ ftfy.fix_text(text, normalization=self.normalization) for text in samples[self.text_key] ] return samples