From 76d594bbbc9a9bb06a14be4af5f2069548992eeb Mon Sep 17 00:00:00 2001 From: James Davies Date: Wed, 13 Nov 2024 15:34:45 +0100 Subject: [PATCH] rebalance halo sampler test buffer size --- src/py21cmfast/wrapper/cfuncs.py | 2 +- tests/test_halo_sampler.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/py21cmfast/wrapper/cfuncs.py b/src/py21cmfast/wrapper/cfuncs.py index 7d724a65..d8c36647 100644 --- a/src/py21cmfast/wrapper/cfuncs.py +++ b/src/py21cmfast/wrapper/cfuncs.py @@ -1203,7 +1203,7 @@ def halo_sample_test( if from_cat: z_prev = (1 + redshift) / global_params.ZPRIME_STEP_FACTOR - 1 - buffer_size = int(2e7) + buffer_size = int(3e7) # about 500MB total 2e7 * 4 (float) * 4 (mass + 3crd) nhalo_out = np.zeros(1).astype("i4") N_out = np.zeros(n_cond).astype("i4") M_out = np.zeros(n_cond).astype("f8") diff --git a/tests/test_halo_sampler.py b/tests/test_halo_sampler.py index 5796a991..f060f3eb 100644 --- a/tests/test_halo_sampler.py +++ b/tests/test_halo_sampler.py @@ -41,7 +41,7 @@ def test_sampler(name, cond, from_cat, plt): from_cat = "cat" in from_cat - n_cond = 25000 + n_cond = 15000 if from_cat: mass = 10 ** options_log10mass[cond] cond = mass