Skip to content

Commit d73900c

Browse files
committed
🐛 fixed missing simulation_kwargs argument in SobolSampler.add_sample
1 parent 8205949 commit d73900c

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

corrai/sampling.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1530,6 +1530,7 @@ def add_sample(
15301530
scramble: bool = True,
15311531
*,
15321532
calc_second_order: bool = True,
1533+
simulation_kwargs: dict = None,
15331534
**sobol_kwargs,
15341535
):
15351536
new_sample = sobol_sampler.sample(
@@ -1539,4 +1540,10 @@ def add_sample(
15391540
calc_second_order=calc_second_order,
15401541
**sobol_kwargs,
15411542
)
1542-
self._post_draw_sample(new_sample, simulate, n_cpu, sample_is_dimless=False)
1543+
self._post_draw_sample(
1544+
new_sample,
1545+
simulate,
1546+
n_cpu,
1547+
sample_is_dimless=False,
1548+
simulation_kwargs=simulation_kwargs,
1549+
)

0 commit comments

Comments
 (0)