Made Gaussian signal a single common noise source for all agents

arnauqb · Aug 3, 2024 · 1e90b96 · 1e90b96
1 parent 81d46b1
commit 1e90b96
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/blackbirds/models/rama_cont.py b/blackbirds/models/rama_cont.py
@@ -35,7 +35,7 @@ def initialize(self, params):
     def step(self, params, x):
         # draw epsilon_t from normal distribution
         sigma = params[2]
-        epsilon_t = torch.distributions.Normal(0, sigma).rsample((self.n_agents,))
+        epsilon_t = torch.distributions.Normal(0, sigma).rsample()
         # compute order
         nu_t = x[-1, 0, :]
         order = self.compute_order(epsilon_t, nu_t)