From 81a958350a39ca23c58fbb28cbffac6c4da09d6a Mon Sep 17 00:00:00 2001 From: Sofie Van Landeghem Date: Sat, 22 Nov 2025 22:38:23 +0100 Subject: [PATCH] do int conversation before taking the max --- nanochat/checkpoint_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nanochat/checkpoint_manager.py b/nanochat/checkpoint_manager.py index afba340..530e6d3 100644 --- a/nanochat/checkpoint_manager.py +++ b/nanochat/checkpoint_manager.py @@ -117,7 +117,7 @@ def find_last_step(checkpoint_dir): checkpoint_files = [f for f in os.listdir(checkpoint_dir) if re.search(r'model_(\d+)\.pt$', f)] if not checkpoint_files: raise FileNotFoundError(f"No checkpoints found in {checkpoint_dir}") - last_step = int(max(f.split("_")[-1].split(".")[0] for f in checkpoint_files)) + last_step = max(int(f.split("_")[-1].split(".")[0]) for f in checkpoint_files) return last_step # -----------------------------------------------------------------------------