Skip to content

Commit fb28789

Browse files
committed
[Fix] Fix unit test of hardpack
1 parent e547e11 commit fb28789

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/datasets/test_dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def test_hard_pack_dataset(random_length, global_pack):
6161

6262
total_input_ids = list(chain(*input_ids_list))
6363

64-
random_gen = random.Random(10)
64+
random_gen = np.random.RandomState(10)
6565
if global_pack:
6666
expected_sample = [[i["input_ids"] for i in d] for d in datasets]
6767
expected_sample = list(chain.from_iterable(expected_sample))

0 commit comments

Comments
 (0)