Skip to content

Commit f406a4d

Browse files
authored
mv test_data_collator.py to test/data (#4843)
* mv test_data_collator.py to test/data * set proper seed 3 for paddle.bernoulli * remove unused comments
1 parent 99eb861 commit f406a4d

File tree

2 files changed

+1
-14
lines changed

2 files changed

+1
-14
lines changed

tests/trainer/test_data_collator.py renamed to tests/data/test_data_collator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,7 @@ def _test_no_pad_and_pad(self, no_pad_features, pad_features):
215215
# Expect error due to padding token missing
216216
data_collator(pad_features)
217217

218-
set_seed(42) # For reproducibility
218+
set_seed(3) # For reproducibility
219219
tokenizer = BertTokenizer(self.vocab_file)
220220
data_collator = DataCollatorForLanguageModeling(tokenizer)
221221
batch = data_collator(no_pad_features)

tests/trainer/__init__.py

Lines changed: 0 additions & 13 deletions
This file was deleted.

0 commit comments

Comments
 (0)