Apply suggestions from code review

Author: Borda

src/litmodels/integrations/checkpoints.py

@@ -57,7 +57,7 @@ def _save_checkpoint(self, trainer: Trainer, filepath: str) -> None:
     return cast(Type[ModelCheckpointType], LitModelCheckpointTemplate)
 
 
-# Create explicit classes with specific names if needed
+# Create explicit classes with specific names
 if _LIGHTNING_AVAILABLE:
     LightningModelCheckpoint = _model_checkpoint_template(LightningModelCheckpoint)
 if _PYTORCHLIGHTNING_AVAILABLE: