fixed reseed and new vel bias

abacusorg · Jan 15, 2024 · 27324cb · 27324cb
1 parent 5eab187
commit 27324cb
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/abacusnbody/hod/abacus_hod.py b/abacusnbody/hod/abacus_hod.py
@@ -564,7 +564,10 @@ def run_hod(self, tracers = None, want_rsd = True, want_nfw = False, NFW_draw =
                 r2 = np.vstack((r20, r21, r22)).T
             r3 = mtg.random(size=len(self.particle_data['prandoms']), nthread=Nthread, dtype=np.float32)
             self.halo_data['hrandoms'] = r1
-            self.halo_data['hveldev'] = r2*self.halo_data['hsigma3d']/np.sqrt(3)
+            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.particle_data['prandoms'] = r3
 
             print("gen randoms took, ", time.time() - start)