fixup! fix: only trigger on_save if the output checkpoint path actually contains files

Signed-off-by: Harikrishnan Balagopal <harikrishmenon@gmail.com>

Author: HarikrishnanBalagopal

tuning/sft_trainer.py

@@ -581,18 +581,32 @@ def save(path: str, trainer: SFTTrainer, tc_callback, log_level="WARNING", args=
         if os.path.exists(path):
             try:
                 saved_files = os.listdir(path)
-                logger.info("sanity check, we found %d files at checkpoint path '%s'", len(saved_files), path)
+                logger.info(
+                    "sanity check, we found %d files at checkpoint path '%s'",
+                    len(saved_files),
+                    path,
+                )
                 actually_saved = len(saved_files) > 0
             except Exception as e:
-                logger.warning("sanity check, failed to list files in checkpoint path '%s' , error: %s", path, e)
+                logger.warning(
+                    "sanity check, failed to list files in checkpoint path '%s' , error: %s",
+                    path,
+                    e,
+                )
         else:
-            logger.warning("sanity check, failed because checkpoint path '%s' doesn't exist", path)
+            logger.warning(
+                "sanity check, failed because checkpoint path '%s' doesn't exist", path
+            )
         if actually_saved:
             tc_callback.on_save(
                 args, trainer.state, trainer.control, path=path, is_final=True
             )
         else:
-            logger.warning("skip triggering on_save event since checkpoint path is empty: '%s'", path)
+            logger.warning(
+                "skip triggering on_save event since checkpoint path is empty: '%s'",
+                path,
+            )
+
 
 def get_parser():
     """Get the command-line argument parser."""