From 4e4d6f6a7f62a1fc72db590188f7e4cad91bf5b9 Mon Sep 17 00:00:00 2001
From: Dan Malkin <danmh@google.com>
Date: Mon, 29 Jul 2024 08:24:03 -0700
Subject: [PATCH] Fixing Issue #1353

PiperOrigin-RevId: 657208154
---
 t5x/checkpoints.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/t5x/checkpoints.py b/t5x/checkpoints.py
index d19165082..13733ec1e 100644
--- 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