filter out tests waiting for next tfp release

numpyro/infer/__init__.py

@@ -1,6 +1,8 @@
 # Copyright Contributors to the Pyro project.
 # SPDX-License-Identifier: Apache-2.0
 
+import warnings
+
 from numpyro.infer.barker import BarkerMH
 from numpyro.infer.elbo import (
     ELBO,
@@ -29,6 +31,11 @@
 
 from . import autoguide, reparam
 
+warnings.filterwarnings(
+    "ignore", message=".*Attempting to hash a tracer.*", category=FutureWarning
+)
+
+
 __all__ = [
     "AIES",
     "autoguide",

pyproject.toml

@@ -88,6 +88,7 @@ known-jax = ["flax", "haiku", "jax", "optax", "tensorflow_probability"]
 addopts = ["-v", "--color=yes"]
 filterwarnings = [
     "error",
+    "ignore:.*Attempting to hash a tracer:FutureWarning",
     "ignore:numpy.ufunc size changed,:RuntimeWarning",
     "ignore:Using a non-tuple sequence:FutureWarning",
     "ignore:jax.tree_structure is deprecated:FutureWarning",

test/contrib/test_tfp.py

@@ -35,6 +35,7 @@ def f(x):
     assert res.scale == 1
 
 
+@pytest.mark.skip(reason="Waiting for the next  tfp release")
 @pytest.mark.filterwarnings("ignore:can't resolve package")
 def test_transformed_distributions():
     from tensorflow_probability.substrates.jax import (
@@ -113,6 +114,7 @@ def make_kernel_fn(target_log_prob_fn):
     )
 
 
+@pytest.mark.skip(reason="Waiting for the next  tfp release")
 @pytest.mark.parametrize(
     "kernel, kwargs",
     [
@@ -243,6 +245,7 @@ def test_sample_tfp_distributions():
 
 # test that sampling from unwrapped tensorflow_probability distributions works as
 # expected using numpyro.sample primitive
+@pytest.mark.skip(reason="Waiting for the next  tfp release")
 @pytest.mark.parametrize(
     "dist,args",
     [
@@ -270,6 +273,7 @@ def test_sample_unwrapped_tfp_distributions(dist, args):
 
 
 # test mixture distributions
+@pytest.mark.skip(reason="Waiting for the next  tfp release")
 def test_sample_unwrapped_mixture_same_family():
     from tensorflow_probability.substrates.jax import distributions as tfd