Disable GeneralizedExtremeValue distribution from `jax_transformati…

…on_test.testLogProbSample`. Re-enable after http://b/175654800 is resolved. PiperOrigin-RevId: 347535479
ENate · Dec 15, 2020 · 90fc5c9 · 90fc5c9
1 parent 7ba09f6
commit 90fc5c9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tensorflow_probability/python/distributions/jax_transformation_test.py b/tensorflow_probability/python/distributions/jax_transformation_test.py
@@ -80,6 +80,7 @@
 
 JVP_SAMPLE_BLOCKLIST = ()
 JVP_LOGPROB_SAMPLE_BLOCKLIST = (
+    'GeneralizedExtremeValue',  # http://b/175654800
     'Skellam',  # http://b/171079052
 )
 JVP_LOGPROB_PARAM_BLOCKLIST = (
@@ -89,6 +90,7 @@
 
 VJP_SAMPLE_BLOCKLIST = ()
 VJP_LOGPROB_SAMPLE_BLOCKLIST = (
+    'GeneralizedExtremeValue',  # http://b/175654800
     'Skellam',  # http://b/171079052
 )
 VJP_LOGPROB_PARAM_BLOCKLIST = (