Fixing Issue #1353

PiperOrigin-RevId: 657208154
google-research · Jul 30, 2024 · 4e4d6f6 · 4e4d6f6
1 parent e1a6876
commit 4e4d6f6
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