[tf.data] Fix logging ops debug statement.
authorjoel-shor <joelshor@google.com>
Mon, 30 Apr 2018 09:20:20 +0000 (12:20 +0300)
committerjoel-shor <joelshor@google.com>
Mon, 30 Apr 2018 09:20:20 +0000 (12:20 +0300)
tensorflow/contrib/data/python/ops/resampling.py

index 47bf6ec..6be4198 100644 (file)
@@ -152,7 +152,7 @@ def _filter_ds(dataset, acceptance_dist_ds, initial_dist_ds, class_values_ds,
     return control_flow_ops.cond(
         math_ops.less(proportion_rejected, .5),
         lambda: accept_dist,
-        lambda: logging_ops.logging.warn(  # pylint: disable=g-long-lambda
+        lambda: logging_ops.Print(  # pylint: disable=g-long-lambda
             accept_dist, [proportion_rejected, initial_dist, accept_dist],
             message="Proportion of examples rejected by sampler is high: ",
             summarize=100,