def augment()

in augmentation/filtering.py [0:0]

• 8 lines of code
• 2 McCabe index (conditional complexity)

    def augment(self, text: str, *args, **kwargs):
        augmented = self.augmenter.augment(text, *args, **kwargs)
        if isinstance(augmented, str):
            augmented = [augmented]
        filter_fn = lambda variant: self.metric_acceptor(
            self.metric_fn(text, variant)
        )
        return list(filter(filter_fn, augmented))