diff options
| author |  Shanqing Cai <cais@google.com> | 2017-11-16 10:23:48 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-16 10:28:33 -0800 |
| commit | aa4162ac9f1812a0966d3cd9b5e441e47f035828 (patch) | |
| tree | 540cd3e4813da0188c3a290b313c802871b31748 /tensorflow/contrib/summary/summary_ops_test.py | |
| parent | 9d737356147a730326cfcbdc08b0b876dd0766e6 (diff) | |
contrib/summary: refactor summary_test_util
A logdir may contain files other than summary event files, e.g., checkpoints.
So add a method "events_from_file" to load events from a single file.
The existing "events_from_logdir" method now calls the new method.
PiperOrigin-RevId: 175981886
Diffstat (limited to 'tensorflow/contrib/summary/summary_ops_test.py')
| -rw-r--r-- | tensorflow/contrib/summary/summary_ops_test.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/summary/summary_ops_test.py b/tensorflow/contrib/summary/summary_ops_test.py index 09169fa6d7..c5ca054f77 100644 --- a/tensorflow/contrib/summary/summary_ops_test.py +++ b/tensorflow/contrib/summary/summary_ops_test.py @@ -79,7 +79,7 @@ class TargetTest(test_util.TensorFlowTestCase): summary_ops.scalar('scalar', 2.0) write() - events = summary_test_util.events_from_file(logdir) + events = summary_test_util.events_from_logdir(logdir) self.assertEqual(len(events), 2) self.assertEqual(events[1].summary.value[0].simple_value, 2.0) @@ -92,7 +92,7 @@ class TargetTest(test_util.TensorFlowTestCase): summary_ops.scalar('scalar', 2.0) - events = summary_test_util.events_from_file(logdir) + events = summary_test_util.events_from_logdir(logdir) self.assertEqual(len(events), 2) self.assertEqual(events[1].summary.value[0].tag, 'scalar') @@ -105,7 +105,7 @@ class TargetTest(test_util.TensorFlowTestCase): summary_ops.scalar('scalar', 2.0, global_step=global_step) - events = summary_test_util.events_from_file(logdir) + events = summary_test_util.events_from_logdir(logdir) self.assertEqual(len(events), 2) self.assertEqual(events[1].summary.value[0].tag, 'scalar') |