fix deprecated nuts and hmc kwargs.

sbi-dev · Apr 4, 2024 · b87f401 · b87f401
1 parent 6e0e98a
commit b87f401
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 8 deletions.
diff --git a/tests/inference_on_device_test.py b/tests/inference_on_device_test.py
@@ -58,7 +58,7 @@
         pytest.param(SNRE_B, "resnet", "slice", marks=pytest.mark.mcmc),
         (SNRE_C, "resnet", "rejection"),
         (SNRE_C, "resnet", "importance"),
-        pytest.param(SNRE_C, "resnet", "nuts", marks=pytest.mark.mcmc),
+        pytest.param(SNRE_C, "resnet", "nuts_pymc", marks=pytest.mark.mcmc),
     ],
 )
 @pytest.mark.parametrize(

diff --git a/tests/linearGaussian_snle_test.py b/tests/linearGaussian_snle_test.py
@@ -406,11 +406,9 @@ def simulator(theta):
         pytest.param("slice_np", "uniform", marks=pytest.mark.mcmc),
         pytest.param("slice_np_vectorized", "gaussian", marks=pytest.mark.mcmc),
         pytest.param("slice_np_vectorized", "uniform", marks=pytest.mark.mcmc),
-        pytest.param("slice", "gaussian", marks=pytest.mark.mcmc),
-        pytest.param("slice", "uniform", marks=pytest.mark.mcmc),
-        pytest.param("nuts", "gaussian", marks=pytest.mark.mcmc),
-        pytest.param("nuts", "uniform", marks=pytest.mark.mcmc),
-        pytest.param("hmc", "gaussian", marks=pytest.mark.mcmc),
+        pytest.param("nuts_pymc", "gaussian", marks=pytest.mark.mcmc),
+        pytest.param("nuts_pyro", "uniform", marks=pytest.mark.mcmc),
+        pytest.param("hmc_pymc", "gaussian", marks=pytest.mark.mcmc),
         ("rejection", "uniform"),
         ("rejection", "gaussian"),
         ("rKL", "uniform"),

diff --git a/tests/linearGaussian_snre_test.py b/tests/linearGaussian_snre_test.py
@@ -328,8 +328,8 @@ def simulator(theta):
         pytest.param("slice_np_vectorized", "uniform", marks=pytest.mark.mcmc),
         pytest.param("slice", "gaussian", marks=pytest.mark.mcmc),
         pytest.param("slice", "uniform", marks=pytest.mark.mcmc),
-        pytest.param("nuts", "gaussian", marks=pytest.mark.mcmc),
-        pytest.param("nuts", "uniform", marks=pytest.mark.mcmc),
+        pytest.param("nuts_pymc", "gaussian", marks=pytest.mark.mcmc),
+        pytest.param("nuts_pyro", "uniform", marks=pytest.mark.mcmc),
         pytest.param("hmc", "gaussian", marks=pytest.mark.mcmc),
         ("rejection", "uniform"),
         ("rejection", "gaussian"),