From 56fd88e04fe489ce71e151bfbb8833a4074b1a69 Mon Sep 17 00:00:00 2001 From: Lutz Roeder Date: Tue, 2 Jul 2024 18:16:56 -0700 Subject: [PATCH] Update TensorFlow script --- tools/tf_script.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tools/tf_script.py b/tools/tf_script.py index 0a0aaaa0b58..5aa54666734 100644 --- a/tools/tf_script.py +++ b/tools/tf_script.py @@ -4,15 +4,24 @@ import logging import os import re +import sys import google.protobuf # pylint: disable=import-error logging.getLogger('tensorflow').setLevel(logging.ERROR) os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' +dup_stderr = os.dup(sys.stderr.fileno()) +null = os.open(os.devnull, os.O_WRONLY) +os.dup2(null, sys.stderr.fileno()) +os.close(null) + from tensorflow.core.framework import api_def_pb2 # pylint: disable=import-error,no-name-in-module,wrong-import-position from tensorflow.core.framework import op_def_pb2 # pylint: disable=import-error,no-name-in-module,wrong-import-position from tensorflow.core.framework import types_pb2 # pylint: disable=import-error,no-name-in-module,wrong-import-position +os.dup2(dup_stderr, sys.stderr.fileno()) +os.close(dup_stderr) + def _read(path): with open(path, 'r', encoding='utf-8') as file: return file.read()