fix reseed bug

abacusorg · Jan 16, 2024 · 54a419c · 54a419c
1 parent 27324cb
commit 54a419c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/abacusnbody/hod/abacus_hod.py b/abacusnbody/hod/abacus_hod.py
@@ -567,7 +567,7 @@ def run_hod(self, tracers = None, want_rsd = True, want_nfw = False, NFW_draw =
             if len(self.halo_data['hveldev'].shape) == 1:
                 self.halo_data['hveldev'] = r20*self.halo_data['hsigma3d']/np.sqrt(3)
             else:
-                self.halo_data['hveldev'] = r2*self.halo_data['hsigma3d']/np.sqrt(3)
+                self.halo_data['hveldev'] = r2*np.repeat(self.halo_data['hsigma3d'], 3).reshape((-1, 3))/np.sqrt(3)
             self.particle_data['prandoms'] = r3
 
             print("gen randoms took, ", time.time() - start)