Internal

Orbax Authors · Orbax Authors · commit fbe611397ba0 · 2026-07-13T15:21:37.000-07:00
PiperOrigin-RevId: 947272180
diff --git a/checkpoint/orbax/checkpoint/_src/checkpointers/async_checkpointer.py b/checkpoint/orbax/checkpoint/_src/checkpointers/async_checkpointer.py
@@ -14,6 +14,7 @@
 
 """AsyncCheckpointer."""
 
+import dataclasses
 import datetime
 import sys
 import threading
diff --git a/checkpoint/orbax/checkpoint/_src/checkpointers/checkpointer.py b/checkpoint/orbax/checkpoint/_src/checkpointers/checkpointer.py
@@ -14,6 +14,7 @@
 
 """Synchronous Checkpointer implementation."""
 
+import dataclasses
 import time
 from typing import Any, Iterable, Optional, Type
 
