def augment()

in augly/text/augmenters/insertion.py [0:0]

• 5 lines of code
• 5 McCabe index (conditional complexity)

    def augment(self, texts: Union[str, List[str]]) -> List[str]:
        texts = texts if isinstance(texts, list) else [texts]

        if self.granularity == "all":
            return [self.insert_chars(text) for text in texts]

        # otherwise, self.granularity == "word"
        return [self.insert_chars_per_word(text) for text in texts]