change max_num_trials to make test pass

sbi-dev · Dec 4, 2023 · 632e48b · 632e48b
1 parent 83a8d8f
commit 632e48b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/embedding_net_test.py b/tests/embedding_net_test.py
@@ -96,7 +96,7 @@ def test_iid_embedding_api(num_trials, num_dim):
 
 
 @pytest.mark.slow
-def test_iid_embedding_varying_num_trials(trial_factor=40, max_num_trials=100):
+def test_iid_embedding_varying_num_trials(trial_factor=50, max_num_trials=20):
     """Test embedding net with varying number of trials."""
     num_dim = 2
     prior = torch.distributions.MultivariateNormal(