!26334 [MD] fix occasional fail in random rampler ut

Merge pull request !26334 from liyong126/fix_random_sampler
2021-11-16 02:49:59 +00:00 · 2021-11-16 02:49:59 +00:00 · 7cdef3da6e
parent 53b736c591 9663884db3
commit 7cdef3da6e
1 changed files with 1 additions and 0 deletions
--- a/tests/ut/python/dataset/test_sampler.py
+++ b/tests/ut/python/dataset/test_sampler.py
@ -48,6 +48,7 @@ def test_sequential_sampler(print_res=False):


 def test_random_sampler(print_res=False):
+    ds.config.set_seed(1234)
    manifest_file = "../data/dataset/testManifestData/test5trainimgs.json"
    map_ = {(172876, 0): 0, (54214, 0): 1, (54214, 1): 2, (173673, 0): 3, (64631, 1): 4}