From 002beb030c7b4c11329594b7deaceb262ad70eff Mon Sep 17 00:00:00 2001 From: Jan Date: Tue, 29 Jul 2025 17:11:48 +0200 Subject: [PATCH] fix failing tarp test --- tests/tarp_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/tarp_test.py b/tests/tarp_test.py index 691b9e66c..ce539e0bb 100644 --- a/tests/tarp_test.py +++ b/tests/tarp_test.py @@ -293,7 +293,7 @@ def test_consistent_run_tarp_results_with_posterior(method): num_dim = 2 prior = BoxUniform(-ones(num_dim), ones(num_dim)) - num_simulations = 1000 + num_simulations = 1500 num_tarp_sims = 500 num_posterior_samples = 1000 @@ -308,7 +308,7 @@ def simulator(theta): theta = prior.sample((num_simulations,)) x = simulator(theta) - inferer.append_simulations(theta, x).train(training_batch_size=200) + inferer.append_simulations(theta, x).train() posterior = inferer.build_posterior() thetas = prior.sample((num_tarp_sims,))