From 73b76a7f6cd18969a8f6379207a63d121aa36a9c Mon Sep 17 00:00:00 2001
From: Michal Januszewski <mjanusz@google.com>
Date: Sat, 21 Dec 2024 13:06:30 -0800
Subject: [PATCH] Internal change.

PiperOrigin-RevId: 708650498
---
 connectomics/jax/checkpoint.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/connectomics/jax/checkpoint.py b/connectomics/jax/checkpoint.py
index 5ef121c..202ec9a 100644
--- a/connectomics/jax/checkpoint.py
+++ b/connectomics/jax/checkpoint.py
@@ -14,7 +14,6 @@
 # limitations under the License.
 """Utilities for model checkpointing."""
 
-import dataclasses
 import re
 from typing import Any, Optional, Sequence, TypeVar