From aa47d4ec3f84f79148b46c4b421c40d86190271d Mon Sep 17 00:00:00 2001
From: Ahmed Ahmed <ahmedah@stanford.edu>
Date: Mon, 28 Oct 2024 02:29:14 -0400
Subject: [PATCH] Update src/levanter/checkpoint.py

Co-authored-by: David Hall <dlwh@cs.stanford.edu>
---
 src/levanter/checkpoint.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/src/levanter/checkpoint.py b/src/levanter/checkpoint.py
index d6f062ffc..651b84c68 100644
--- a/src/levanter/checkpoint.py
+++ b/src/levanter/checkpoint.py
@@ -27,7 +27,6 @@
 
 from levanter.tensorstore_serialization import tree_deserialize_leaves_tensorstore, tree_serialize_leaves_tensorstore
 from levanter.types import FilterSpec
-# from levanter.trainer import StepInfo
 
 
 logger = logging.getLogger(__name__)