Merge pull request #521 from dangne/master

Fix a bug when running textattack eval with --num-examples=-1
QData · Sep 24, 2021 · e7a824e · e7a824e
2 parents cdfcba4 + 19148ea
commit e7a824e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/textattack/commands/eval_model_command.py b/textattack/commands/eval_model_command.py
@@ -39,6 +39,8 @@ def get_preds(self, model, inputs):
     def test_model_on_dataset(self, args):
         model = ModelArgs._create_model_from_args(args)
         dataset = DatasetArgs._create_dataset_from_args(args)
+        if args.num_examples == -1:
+            args.num_examples = len(dataset)
 
         preds = []
         ground_truth_outputs = []