fix random seed for subsampling instruction so that it doesnt fail

uio-bmi · Apr 25, 2024 · ae3be16 · ae3be16
1 parent df535e4
commit ae3be16
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/test/workflows/instructions/test_SubsamplingInstruction.py b/test/workflows/instructions/test_SubsamplingInstruction.py
@@ -1,4 +1,5 @@
 import os
+import random
 import shutil
 from unittest import TestCase
 
@@ -11,6 +12,7 @@
 
 class TestSubsamplingInstruction(TestCase):
     def test_run(self):
+        random.seed(1)
         path = PathBuilder.build(EnvironmentSettings.tmp_test_path / "subsampling/")
         dataset = RandomDatasetGenerator.generate_receptor_dataset(200, labels={"epitope": {"A": 0.5, "B": 0.5}}, path=path,
                                                                    chain_1_length_probabilities={3: 1}, chain_2_length_probabilities={4: 1})