We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e547e11 commit fb28789Copy full SHA for fb28789
1 file changed
tests/datasets/test_dataset.py
@@ -61,7 +61,7 @@ def test_hard_pack_dataset(random_length, global_pack):
61
62
total_input_ids = list(chain(*input_ids_list))
63
64
- random_gen = random.Random(10)
+ random_gen = np.random.RandomState(10)
65
if global_pack:
66
expected_sample = [[i["input_ids"] for i in d] for d in datasets]
67
expected_sample = list(chain.from_iterable(expected_sample))
0 commit comments