diff --git a/tests/callbacks/test_rich_progress_bar.py b/tests/callbacks/test_rich_progress_bar.py index c6f44759ba371..10ff0f7cd9ea6 100644 --- a/tests/callbacks/test_rich_progress_bar.py +++ b/tests/callbacks/test_rich_progress_bar.py @@ -17,7 +17,7 @@ from pytorch_lightning import Trainer from pytorch_lightning.callbacks import ProgressBarBase, RichProgressBar -from tests.helpers.boring_model import BoringModel +from pytorch_lightning.utilities.debugging_examples import BoringModel from tests.helpers.runif import RunIf