add deterministic test seed to easy networks tests

cnellington · Apr 21, 2024 · 010ebfc · 010ebfc
1 parent 38cc018
commit 010ebfc
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/contextualized/easy/tests.py b/contextualized/easy/tests.py
@@ -46,6 +46,8 @@ def setUp(self):
         """
         Shared unit test setup code.
         """
+        np.random.seed(0)
+        torch.manual_seed(0)
         self.n_samples = 100
         self.c_dim = 4
         self.x_dim = 5
@@ -81,6 +83,8 @@ def setUp(self):
         """
         Shared unit test setup code.
         """
+        np.random.seed(0)
+        torch.manual_seed(0)
         self.n_samples = 100
         self.c_dim = 4
         self.x_dim = 5
@@ -124,6 +128,8 @@ def setUp(self):
         """
         Shared unit test setup code.
         """
+        np.random.seed(0)
+        torch.manual_seed(0)
         self.n_samples = 100
         self.c_dim = 4
         self.x_dim = 5