fix #1224: upgrade pyknos, remove xfail test.

sbi-dev · Aug 20, 2024 · 72b3a7d · 72b3a7d
1 parent ac29077
commit 72b3a7d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 14 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -35,7 +35,7 @@ dependencies = [
     "matplotlib",
     "numpy",
     "pillow",
-    "pyknos>=0.15.1",
+    "pyknos>=0.16.0",
     "pyro-ppl>=1.3.1",
     "scikit-learn",
     "scipy",

diff --git a/tests/posterior_nn_test.py b/tests/posterior_nn_test.py
@@ -213,19 +213,7 @@ def test_batched_mcmc_sample_log_prob_with_different_x(
 
 
 @pytest.mark.slow
-@pytest.mark.parametrize(
-    "density_estimator",
-    [
-        pytest.param(
-            "mdn",
-            marks=pytest.mark.xfail(
-                raises=AssertionError, reason="Due to MDN bug in pyknos", strict=True
-            ),
-        ),
-        "maf",
-        "zuko_nsf",
-    ],
-)
+@pytest.mark.parametrize("density_estimator", ["mdn", "maf", "zuko_nsf"])
 def test_batched_sampling_and_logprob_accuracy(density_estimator: str):
     """Test with two different observations and compare to sequential methods."""
     num_dim = 2