diff --git a/tensorflow/python/training/monitored_session.py b/tensorflow/python/training/monitored_session.py index 2feb6bd3c3f..383a9d1e30a 100644 --- a/tensorflow/python/training/monitored_session.py +++ b/tensorflow/python/training/monitored_session.py @@ -535,7 +535,7 @@ class _MonitoredSession(object): # __exit__ should return True to suppress an exception. return exception_type is None - class _CoordinatedSessionCreator(object): + class _CoordinatedSessionCreator(SessionCreator): """Factory for the _RecoverableSession.""" def __init__(self, session_creator, hooks, stop_grace_period_secs):