From 8be2cc514926868cbb0a937f0d86fb18243ec39f Mon Sep 17 00:00:00 2001
From: Mihai Maruseac <mihaimaruseac@google.com>
Date: Tue, 22 Dec 2020 15:23:21 -0800
Subject: [PATCH] Print directly

---
 .../python/kernel_tests/boosted_trees/resource_ops_test.py  | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py b/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py
index a331b8670f7..250d5a403fc 100644
--- a/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py
+++ b/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py
@@ -16,10 +16,8 @@
 from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
-__p = print
-def _p(msg):
-  import sys
-  __p(msg, file=sys.stderr)
+import sys
+_p = print
 
 from google.protobuf import text_format