diff --git a/tests/tests_pytorch/tuner/test_lr_finder.py b/tests/tests_pytorch/tuner/test_lr_finder.py index 442c795491320..e2d1b6bd4ee84 100644 --- a/tests/tests_pytorch/tuner/test_lr_finder.py +++ b/tests/tests_pytorch/tuner/test_lr_finder.py @@ -599,7 +599,9 @@ def configure_optimizers(self): # Verify learning rate finder ran and has results assert lr_finder_callback.optimal_lr is not None, "Learning rate finder should have results" - assert lr_finder_callback.optimal_lr.suggestion() > 0, "Learning rate suggestion should be positive" + suggestion = lr_finder_callback.optimal_lr.suggestion() + if suggestion is not None: + assert suggestion > 0, "Learning rate suggestion should be positive" def test_gradient_correctness():