add interactive mode + test

2021-10-13 00:05:06 +03:00 · 2020-05-19 16:12:24 -04:00
parent ed9857608a
commit d0e54457c0
4 changed files with 77 additions and 5 deletions
--- a/textattack/shared/attack.py
+++ b/textattack/shared/attack.py
@@ -162,6 +162,8 @@ class Attack:
        
        if shuffle:
            random.shuffle(dataset.examples)
+        
+        num_examples = num_examples or len(dataset)        

        if num_examples <= 0:
            return