explicitly seed multiagent environment

cpnota · Jan 26, 2024 · e236f36 · e236f36
1 parent 109a7a1
commit e236f36
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/all/experiments/multiagent_env_experiment_test.py b/all/experiments/multiagent_env_experiment_test.py
@@ -20,7 +20,7 @@ def setUp(self):
         random.seed(0)
         np.random.seed(0)
         torch.manual_seed(0)
-        self.env = MultiagentAtariEnv('space_invaders_v2', device='cpu')
+        self.env = MultiagentAtariEnv('space_invaders_v2', device='cpu', seed=0)
         self.env.reset(seed=0)
         self.experiment = None