From 1e548c4db7865f8bbbc3bd374ec6e0ebf22a4a93 Mon Sep 17 00:00:00 2001
From: Katherine Wu <kathywu@google.com>
Date: Fri, 13 Nov 2020 08:55:39 -0800
Subject: [PATCH] Suppress warnings when listing functions for serialization.
PiperOrigin-RevId: 342271395
Change-Id: Ieedeeb7817f794c41dae3eb06b9b0d3b607fa89a
---
tensorflow/python/training/tracking/tracking.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/tensorflow/python/training/tracking/tracking.py b/tensorflow/python/training/tracking/tracking.py
index 3abafdcb233..744db32c550 100644
--- a/tensorflow/python/training/tracking/tracking.py
+++ b/tensorflow/python/training/tracking/tracking.py
@@ -18,6 +18,7 @@ from __future__ import division
from __future__ import print_function
import copy
+import warnings
from absl import logging
@@ -107,7 +108,9 @@ class AutoTrackable(base.Trackable):
logging_verbosity = logging.get_verbosity()
try:
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
# We really don't want to throw an exception just because some object's
# attribute accessor is broken.