fix: tsnpe sir test

sbi-dev · Jul 30, 2024 · fc434b7 · fc434b7
1 parent b275448
commit fc434b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/linearGaussian_snpe_test.py b/tests/linearGaussian_snpe_test.py
@@ -356,7 +356,7 @@ def simulator(theta):
     elif method_str.startswith("tsnpe"):
         sample_method = "rejection" if method_str == "tsnpe_rejection" else "sir"
         inference = SNPE_C(**creation_args)
-        theta = prior.sample((900,))
+        theta = prior.sample((1000,))
         x = simulator(theta)
         posterior_estimator = inference.append_simulations(theta, x).train()
         posterior1 = DirectPosterior(