From 9e5bbd98197461d816f9b85d76b59758ffdeaacd Mon Sep 17 00:00:00 2001
From: Lei Lei <18294546+Crescent-Saturn@users.noreply.github.com>
Date: Fri, 8 Nov 2024 17:44:39 -0500
Subject: [PATCH] fix: saving ckpt error when ckpt folder has some folder
 inside (#3516)

---
 nerfstudio/engine/trainer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/nerfstudio/engine/trainer.py b/nerfstudio/engine/trainer.py
index cac31cf50..d27185526 100644
--- a/nerfstudio/engine/trainer.py
+++ b/nerfstudio/engine/trainer.py
@@ -478,8 +478,8 @@ class Trainer:
         )
         # possibly delete old checkpoints
         if self.config.save_only_latest_checkpoint:
-            # delete everything else in the checkpoint folder
-            for f in self.checkpoint_dir.glob("*"):
+            # delete every other checkpoint in the checkpoint folder
+            for f in self.checkpoint_dir.glob("*.ckpt"):
                 if f != ckpt_path:
                     f.unlink()