diff options
| author |  Pavel Sountsov <siege@google.com> | 2018-10-09 11:56:25 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-09 12:03:38 -0700 |
| commit | 8c2a52b26f21167ed0fcec7859850e38d0c216f9 (patch) | |
| tree | af808bef14df5ab8d0c0c0b2172d695c5f194ae1 /tensorflow/python | |
| parent | 072fcb995a3fd658ee2461b59b159498c710513d (diff) | |
Silence tf.distributions deprecation messages caused by internal global function calls.
E.g. register_kl calls would trigger such warnings. This spam was exacerbated
by the fact that it happens before logging is initialized, so it is dumped
prominently to STDERR. Worse yet it also happened no matter whether the user
imported any symbols from tf.distributions or not as the relevant code is
executed when you import TensorFlow.
PiperOrigin-RevId: 216396036
Diffstat (limited to 'tensorflow/python')
| -rw-r--r-- | tensorflow/python/ops/distributions/distributions.py | 35 |
1 files changed, 19 insertions, 16 deletions
diff --git a/tensorflow/python/ops/distributions/distributions.py b/tensorflow/python/ops/distributions/distributions.py index 59ed455e43..b18caa5b2e 100644 --- a/tensorflow/python/ops/distributions/distributions.py +++ b/tensorflow/python/ops/distributions/distributions.py @@ -17,21 +17,24 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function +from tensorflow.python.util import deprecation -# pylint: disable=wildcard-import,unused-import -from tensorflow.python.ops.distributions.bernoulli import Bernoulli -from tensorflow.python.ops.distributions.beta import Beta -from tensorflow.python.ops.distributions.categorical import Categorical -from tensorflow.python.ops.distributions.dirichlet import Dirichlet -from tensorflow.python.ops.distributions.dirichlet_multinomial import DirichletMultinomial -from tensorflow.python.ops.distributions.distribution import * -from tensorflow.python.ops.distributions.exponential import Exponential -from tensorflow.python.ops.distributions.gamma import Gamma -from tensorflow.python.ops.distributions.kullback_leibler import * -from tensorflow.python.ops.distributions.laplace import Laplace -from tensorflow.python.ops.distributions.multinomial import Multinomial -from tensorflow.python.ops.distributions.normal import Normal -from tensorflow.python.ops.distributions.student_t import StudentT -from tensorflow.python.ops.distributions.uniform import Uniform -# pylint: enable=wildcard-import,unused-import +# pylint: disable=wildcard-import,unused-import,g-import-not-at-top +with deprecation.silence(): + from tensorflow.python.ops.distributions.bernoulli import Bernoulli + from tensorflow.python.ops.distributions.beta import Beta + from tensorflow.python.ops.distributions.categorical import Categorical + from tensorflow.python.ops.distributions.dirichlet import Dirichlet + from tensorflow.python.ops.distributions.dirichlet_multinomial import DirichletMultinomial + from tensorflow.python.ops.distributions.distribution import * + from tensorflow.python.ops.distributions.exponential import Exponential + from tensorflow.python.ops.distributions.gamma import Gamma + from tensorflow.python.ops.distributions.kullback_leibler import * + from tensorflow.python.ops.distributions.laplace import Laplace + from tensorflow.python.ops.distributions.multinomial import Multinomial + from tensorflow.python.ops.distributions.normal import Normal + from tensorflow.python.ops.distributions.student_t import StudentT + from tensorflow.python.ops.distributions.uniform import Uniform +# pylint: enable=wildcard-import,unused-import +del deprecation |