shorten test_train

QData · Sep 11, 2023 · 40e4686 · 40e4686
1 parent 6c1d2f8
commit 40e4686
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_command_line/test_train.py b/tests/test_command_line/test_train.py
@@ -5,7 +5,7 @@
 
 
 def test_train_tiny():
-    command = "textattack train --model distilbert-base-uncased --attack textfooler --dataset rotten_tomatoes --model-max-length 64  --num-epochs 1 --num-clean-epochs 0 --num-train-adv-examples 5"
+    command = "textattack train --model distilbert-base-uncased --attack textfooler --dataset rotten_tomatoes --model-max-length 64  --num-epochs 1 --num-clean-epochs 0 --num-train-adv-examples 2"
 
     # Run command and validate outputs.
     result = run_command_and_get_result(command)