diff --git a/docs/hod.rst b/docs/hod.rst index 5435afe..b3b7dcb 100644 --- a/docs/hod.rst +++ b/docs/hod.rst @@ -188,7 +188,7 @@ there. Here we show the scripts within ``run_hod.py`` for reference.:: newBall = AbacusHOD(sim_params, HOD_params, clustering_params) # first hod run, slow due to compiling jit, write to disk - mock_dict = newBall.run_hod(newBall.tracers, want_rsd, write_to_disk, Nthread = 16) + mock_dict = newBall.run_hod(newBall.tracers, want_rsd, write_to_disk = write_to_disk, Nthread = 16) # run the 10 different HODs for timing for i in range(10): @@ -207,7 +207,7 @@ redshift-space 2PCF (:math:`\xi(r_p, \pi)`): :: pimax = clustering_params['pimax'] pi_bin_size = clustering_params['pi_bin_size'] # the pi binning is configrured by pi_max and bin size - mock_dict = newBall.run_hod(newBall.tracers, want_rsd, write_to_disk) + mock_dict = newBall.run_hod(newBall.tracers, want_rsd, write_to_disk=write_to_disk) xirppi = newBall.compute_xirppi(mock_dict, rpbins, pimax, pi_bin_size) Light Cones diff --git a/docs/tutorials/analysis/zcv.ipynb b/docs/tutorials/analysis/zcv.ipynb index e1c652d..65bdc1f 100644 --- a/docs/tutorials/analysis/zcv.ipynb +++ b/docs/tutorials/analysis/zcv.ipynb @@ -160,7 +160,9 @@ "source": [ "# Run hod\n", "newBall = AbacusHOD(sim_params, HOD_params, clustering_params)\n", - "mock_dict = newBall.run_hod(newBall.tracers, want_rsd, write_to_disk, Nthread=16)\n", + "mock_dict = newBall.run_hod(\n", + " newBall.tracers, want_rsd, write_to_disk=write_to_disk, Nthread=16\n", + ")\n", "nobj = mock_dict['LRG']['mass'].size\n", "print('number of galaxies', nobj)" ] diff --git a/scripts/hod/run_hod.py b/scripts/hod/run_hod.py index 7944862..06e3d4b 100755 --- a/scripts/hod/run_hod.py +++ b/scripts/hod/run_hod.py @@ -41,7 +41,7 @@ def main(path2config): # throw away run for jit to compile, write to disk mock_dict = newBall.run_hod( - newBall.tracers, want_rsd, write_to_disk=False, Nthread=16 + newBall.tracers, want_rsd, write_to_disk=write_to_disk, Nthread=16 ) # mock_dict = newBall.gal_reader() start = time.time() @@ -61,7 +61,7 @@ def main(path2config): # print("alpha = ",newBall.tracers['LRG']['alpha']) start = time.time() mock_dict = newBall.run_hod( - newBall.tracers, want_rsd, write_to_disk, Nthread=64 + newBall.tracers, want_rsd, write_to_disk=False, Nthread=64 ) print('Done hod, took time ', time.time() - start) start = time.time()