Skip to content

Commit 40e4686

Browse files
committed
shorten test_train
1 parent 6c1d2f8 commit 40e4686

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_command_line/test_train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66

77
def test_train_tiny():
8-
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"
8+
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"
99

1010
# Run command and validate outputs.
1111
result = run_command_and_get_result(command)

0 commit comments

Comments
 (0)