change raise exception to raise

Co-authored-by: Carlos Mocholí <[email protected]>

Author: daniellepintz

pytorch_lightning/trainer/trainer.py

@@ -522,7 +522,7 @@ def _call_and_handle_interrupt(self, trainer_fn: Callable, *args: Any, **kwargs:
             # reset bookkeeping
             self.state.stage = None
             self.on_exception(exception)
-            raise exception
+            raise
 
     def fit(
         self,