Make CI work again (#1611)

pyro-ppl · Jun 20, 2023 · 1b787ae · 1b787ae
1 parent 0e50bac
commit 1b787ae
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -2,6 +2,7 @@ numpyro.egg-info
 __pycache__/
 .ipynb_checkpoints/
 build
+.DS_Store
 
 # built / compiled
 *.pyc

diff --git a/numpyro/contrib/nested_sampling.py b/numpyro/contrib/nested_sampling.py
@@ -260,7 +260,6 @@ def prior_model() -> PriorModelGen:
             num_live_points=model.U_ndims * 25,
             num_parallel_samplers=1,
             max_samples=1e4,
-            uncert_improvement_patience=2,
         )
         default_termination_kwargs = dict(live_evidence_frac=1e-4)
         # Fill-in missing values with defaults. This allows user to inspect what was actually used by inspecting

diff --git a/numpyro/infer/autoguide.py b/numpyro/infer/autoguide.py
@@ -1322,7 +1322,7 @@ def fn(x):
                 data=global_latents
             ):
                 global_outputs = self.global_guide.model(*args, **kwargs)
-            local_args = (global_ouputs,)
+            local_args = (global_outputs,)
             local_kwargs = {}
         else:
             global_latents = {}

diff --git a/test/contrib/test_nested_sampling.py b/test/contrib/test_nested_sampling.py
@@ -85,7 +85,7 @@ def model():
             "x", dist.MultivariateNormal(jnp.zeros(2), covariance_matrix=true_cov)
         )
 
-    ns = NestedSampler(model)
+    ns = NestedSampler(model, constructor_kwargs={"num_live_points": 200})
     ns.run(random.PRNGKey(0))
 
     samples = ns.get_samples(random.PRNGKey(1), 1000)["x"]