change datatype of on_exception hook from Exception to BaseException in ModelCheckpoint

Author: vsey

src/lightning/pytorch/callbacks/model_checkpoint.py

@@ -342,7 +342,7 @@ def on_validation_end(self, trainer: "pl.Trainer", pl_module: "pl.LightningModul
             self._save_last_checkpoint(trainer, monitor_candidates)
 
     @override
-    def on_exception(self, trainer: "pl.Trainer", pl_module: "pl.LightningModule", exception: Exception) -> None:
+    def on_exception(self, trainer: "pl.Trainer", pl_module: "pl.LightningModule", exception: BaseException) -> None:
         """Save a checkpoint when an exception is raised."""
         if not self._should_save_on_exception(trainer):
             return