Fixing Issue #1353

PiperOrigin-RevId: 657208154
google-research · Jul 29, 2024 · 0255754 · 0255754
1 parent f17e82f
commit 0255754
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/t5x/checkpoints.py b/t5x/checkpoints.py
@@ -62,6 +62,9 @@
 import tensorstore as ts
 import typing_extensions
 
+from tensorboard.backend.event_processing import directory_watcher
+from tensorboard.backend.event_processing import event_file_loader
+from tensorboard.backend.event_processing import io_wrapper
 
 PartitionSpec = partitioning.PartitionSpec
 PyTree = Any