Do not assert session logs for training with supervisor, thread timing is

non-deterministic.
Change: 125689946

1 files changed, 4 insertions, 3 deletions

diff --git a/tensorflow/contrib/testing/python/framework/fake_summary_writer.py b/tensorflow/contrib/testing/python/framework/fake_summary_writer.py
index 8cc2c4c464..d742608cf1 100644
--- a/tensorflow/contrib/testing/python/framework/fake_summary_writer.py
+++ b/tensorflow/contrib/testing/python/framework/fake_summary_writer.py
@@ -71,9 +71,10 @@ class FakeSummaryWriter(object):
           if 'global_step/sec' != v.tag:
             actual_simple_values[v.tag] = v.simple_value
       test_case.assertEqual(expected_summaries[step], actual_simple_values)
-    test_case.assertEqual(expected_added_graphs or [], self._added_graphs)
-    test_case.assertEqual(
-        expected_session_logs or [], self._added_session_logs)
+    if expected_added_graphs is not None:
+      test_case.assertEqual(expected_added_graphs, self._added_graphs)
+    if expected_session_logs is not None:
+      test_case.assertEqual(expected_session_logs, self._added_session_logs)
 
   def add_summary(self, summary, current_global_step):
     """Add summary."""