diff --git a/abacusnbody/hod/abacus_hod.py b/abacusnbody/hod/abacus_hod.py index 7f21716..0571cfb 100644 --- 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)