Browse Source

Fix seed in test

Matthias Reso 1 year ago
parent
commit
97a7871f4b
1 changed files with 1 additions and 0 deletions
  1. 1 0
      tests/test_length_based_batch_sampler.py

+ 1 - 0
tests/test_length_based_batch_sampler.py

@@ -12,6 +12,7 @@ SAMPLES = 33
 
 @pytest.fixture
 def dataset():
+    random.seed(42)
     dataset = []
     def add_samples(ds, n, a, b):
         for _ in range(n):