[docs]@OPERATORS.register_module('clean_email_mapper')classCleanEmailMapper(Mapper):"""Mapper to clean email in text samples."""_batched_op=True
[docs]def__init__(self,pattern:Optional[str]=None,repl:str='',*args,**kwargs):""" Initialization method. :param pattern: regular expression pattern to search for within text. :param repl: replacement string, default is empty string. :param args: extra args :param kwargs: extra args """super().__init__(*args,**kwargs)ifpatternisNone:self.pattern=r'[A-Za-z0-9.\-+_]+@[a-z0-9.\-+_]+\.[a-z]+'else:self.pattern=patternif((len(pattern)>2)and(pattern.startswith("r'")andpattern.endswith("'")orpattern.startswith('r"')andpattern.endswith('"'))):self.pattern=pattern[2:-1]self.repl=repl