Merge pull request tensorflow#236 from ywkim:filterwarnings

PiperOrigin-RevId: 227751751
darthdeus · Jan 3, 2019 · 70253ba · 70253ba
2 parents d77c122 + 05b97e8
commit 70253ba
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/tensorflow_probability/python/mcmc/metropolis_hastings.py b/tensorflow_probability/python/mcmc/metropolis_hastings.py
@@ -35,8 +35,9 @@
 
 # Cause all warnings to always be triggered.
 # Not having this means subsequent calls wont trigger the warning.
-warnings.simplefilter('always')
-
+warnings.filterwarnings('always',
+                        module='tensorflow_probability.*metropolis_hastings',
+                        append=True)  # Don't override user-set filters.
 
 MetropolisHastingsKernelResults = collections.namedtuple(
     'MetropolisHastingsKernelResults',

diff --git a/tensorflow_probability/python/mcmc/sample.py b/tensorflow_probability/python/mcmc/sample.py
@@ -34,7 +34,9 @@
 
 # Cause all warnings to always be triggered.
 # Not having this means subsequent calls wont trigger the warning.
-warnings.simplefilter("always")
+warnings.filterwarnings("always",
+                        module="tensorflow_probability.*sample",
+                        append=True)  # Don't override user-set filters.
 
 
 def sample_chain(

diff --git a/tensorflow_probability/python/util/variables.py b/tensorflow_probability/python/util/variables.py
@@ -29,7 +29,9 @@
 
 # Cause all warnings to always be triggered.
 # Not having this means subsequent calls wont trigger the warning.
-warnings.simplefilter("always")
+warnings.filterwarnings("always",
+                        module="tensorflow_probability.*variables",
+                        append=True)  # Don't override user-set filters.
 
 
 def externalize_variables_as_args(fn,