Skip to content

Commit d20c118

Browse files
authored
Add summary source metadata to the TB summary writer output (#6002)
Specifies source metadata for event files as `tensorboard.summary.writer.event_file_writer` for the TB summary writer.
1 parent 3c0a013 commit d20c118

File tree

3 files changed

+13
-1
lines changed

3 files changed

+13
-1
lines changed

tensorboard/summary/writer/event_file_writer.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,11 @@ def __init__(
9090

9191
# Initialize an event instance.
9292
_event = event_pb2.Event(
93-
wall_time=time.time(), file_version="brain.Event:2"
93+
wall_time=time.time(),
94+
file_version="brain.Event:2",
95+
source_metadata=event_pb2.SourceMetadata(
96+
writer="tensorboard.summary.writer.event_file_writer"
97+
),
9498
)
9599
self.add_event(_event)
96100
self.flush()

tensorboard/summary/writer/event_file_writer_fsspec_test.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,10 @@ def test_async_writer_without_write(self):
8181
r.GetNext()
8282
s = event_pb2.Event.FromString(r.record())
8383
self.assertEqual(s.file_version, "brain.Event:2")
84+
self.assertEqual(
85+
s.source_metadata.writer,
86+
"tensorboard.summary.writer.event_file_writer",
87+
)
8488

8589

8690
if __name__ == "__main__":

tensorboard/summary/writer/event_file_writer_test.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,10 @@ def test_async_writer_without_write(self):
6464
r.GetNext()
6565
s = event_pb2.Event.FromString(r.record())
6666
self.assertEqual(s.file_version, "brain.Event:2")
67+
self.assertEqual(
68+
s.source_metadata.writer,
69+
"tensorboard.summary.writer.event_file_writer",
70+
)
6771

6872

6973
class AsyncWriterTest(tb_test.TestCase):

0 commit comments

Comments
 (0)