diff --git a/textattack/shared/attack.py b/textattack/shared/attack.py index 6f390615..3c8a2f1a 100644 --- a/textattack/shared/attack.py +++ b/textattack/shared/attack.py @@ -77,7 +77,9 @@ class Attack: # Give search method access to functions for getting transformations and evaluating them self.search_method.get_transformations = self.get_transformations - self.search_method.get_goal_results = self.goal_function.get_results_from_search_method + self.search_method.get_goal_results = ( + self.goal_function.get_results_from_search_method + ) self.search_method.filter_transformations = self.filter_transformations def get_transformations(self, current_text, original_text=None, **kwargs):