fixed particles initialisation in random_generators

xsuite · Feb 20, 2023 · 5840adf · 5840adf
1 parent 162e834
commit 5840adf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/checkpoint_restart.dat b/checkpoint_restart.dat
diff --git a/xtrack/random/random_generators.py b/xtrack/random/random_generators.py
@@ -48,7 +48,8 @@ def generate(self, n_samples=1000, n_seeds=1000, particles=None):
         context = self._context
         n_seeds = int(n_seeds)
         if particles is None:
-            particles = xp.Particles(state=np.ones(n_seeds), _context=context)
+            particles = xp.Particles(state=np.ones(n_seeds),
+                                     x=np.ones(n_seeds), _context=context)
         elif n_seeds != len(particles._rng_s1):
             print("Warning: both 'particles' and 'n_seeds' are given, but "
                   + "are not compatible. Ignoring 'n_seeds'...")