experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Suppress warnings when listing functions for serialization.

Browse Source 
PiperOrigin-RevId: 342271395
Change-Id: Ieedeeb7817f794c41dae3eb06b9b0d3b607fa89a
This commit is contained in:
Katherine Wu 2020-11-13 08:55:39 -08:00 committed by Geeta Chavan
parent 6eee4b02a8
commit 1e548c4db7
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/python/training/tracking/tracking.py
View File

@ -18,6 +18,7 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import copy import copy
import warnings
from absl import logging from absl import logging
@ -107,7 +108,9 @@ class AutoTrackable(base.Trackable):
logging_verbosity = logging.get_verbosity() logging_verbosity = logging.get_verbosity()
try: try:
logging.set_verbosity(logging.FATAL) logging.set_verbosity(logging.FATAL)
attribute_value = getattr(self, attribute_name, None) with warnings.catch_warnings():
warnings.simplefilter("ignore")
attribute_value = getattr(self, attribute_name, None)
except Exception: # pylint: disable=broad-except except Exception: # pylint: disable=broad-except
# We really don't want to throw an exception just because some object's # We really don't want to throw an exception just because some object's
# attribute accessor is broken. # attribute accessor is broken.