From 8c2a52b26f21167ed0fcec7859850e38d0c216f9 Mon Sep 17 00:00:00 2001 From: Pavel Sountsov Date: Tue, 9 Oct 2018 11:56:25 -0700 Subject: 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 --- .../python/ops/distributions/distributions.py | 35 ++++++++++++---------- 1 file changed, 19 insertions(+), 16 deletions(-) (limited to 'tensorflow/python') 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 -- cgit v1.2.3