clean test_torch and set seed

alan-turing-institute · Mar 13, 2024 · 59ee578 · 59ee578
1 parent 0c74bf2
commit 59ee578
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_torch.py b/tests/test_torch.py
@@ -5,8 +5,6 @@
 from autoemulate.emulators import NeuralNetTorch
 from autoemulate.utils import set_random_seed
 
-set_random_seed(1234)
-
 
 def test_nn_torch_initialisation():
     nn_torch = NeuralNetTorch()
@@ -119,6 +117,7 @@ def test_nn_torch_module_ui():
 
 
 def test_nn_torch_module_ui_param_search_random():
+    set_random_seed(1234)
     input_size, output_size = 10, 2
     X = np.random.rand(100, input_size)
     y = np.random.rand(100, output_size)
@@ -136,6 +135,7 @@ def test_nn_torch_module_ui_param_search_random():
 
 
 def test_nn_torch_module_ui_param_search_bayes():
+    set_random_seed(1234)
     input_size, output_size = 10, 2
     X = np.random.rand(10, input_size)
     y = np.random.rand(10, output_size)