Merge pull request #602 from zjzh/master

Update checkpoint.py
QData · Feb 20, 2022 · f0f0779 · f0f0779
2 parents 1cc6509 + 1d06c65
commit f0f0779
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/textattack/shared/checkpoint.py b/textattack/shared/checkpoint.py
@@ -233,10 +233,8 @@ def _verify(self):
             self.worklist
         ), "Recorded number of remaining attacks and size of worklist are different."
 
-        results_set = set()
-        for result in self.attack_log_manager.results:
-            results_set.add(result.original_text)
-
+        results_set = {result.original_text 
+                       for result in self.attack_log_manager.results}
         assert (
             len(results_set) == self.results_count
         ), "Duplicate `AttackResults` found."