Set random seed for unit tests

ML4GW · Feb 5, 2025 · 8cb41a0 · 8cb41a0
1 parent 6f0514e
commit 8cb41a0
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -1,9 +1,19 @@
+import random
+
 import numpy as np
 import pytest
 import torch
 from scipy.special import erfinv
 
 
+@pytest.fixture(autouse=True)
+def seed_everything():
+    seed = 101589
+    torch.manual_seed(seed)
+    np.random.seed(seed)
+    random.seed(seed)
+
+
 @pytest.fixture
 def compare_against_numpy():
     """