fix failing tests

Author: daniellepintz

pytorch_lightning/callbacks/lambda_function.py

@@ -75,6 +75,7 @@ def __init__(
         on_test_start: Optional[Callable] = None,
         on_test_end: Optional[Callable] = None,
         on_keyboard_interrupt: Optional[Callable] = None,
+        on_exception: Optional[Callable] = None,
         on_save_checkpoint: Optional[Callable] = None,
         on_load_checkpoint: Optional[Callable] = None,
         on_before_backward: Optional[Callable] = None,

pytorch_lightning/trainer/connectors/logger_connector/fx_validator.py

@@ -62,6 +62,7 @@ class FxValidator:
         on_predict_batch_start=None,
         on_predict_batch_end=None,
         on_keyboard_interrupt=None,
+        on_exception=None,
         on_save_checkpoint=None,
         on_load_checkpoint=None,
         setup=None,

tests/trainer/logging_/test_logger_connector.py

@@ -49,6 +49,7 @@ def test_fx_validator(tmpdir):
         "on_init_end",
         "on_init_start",
         "on_keyboard_interrupt",
+        "on_exception",
         "on_load_checkpoint",
         "on_pretrain_routine_end",
         "on_pretrain_routine_start",
@@ -91,6 +92,7 @@ def test_fx_validator(tmpdir):
         "on_init_end",
         "on_init_start",
         "on_keyboard_interrupt",
+        "on_exception",
         "on_load_checkpoint",
         "on_pretrain_routine_end",
         "on_pretrain_routine_start",