import

Author: Borda

src/lightning/pytorch/callbacks/progress/rich_progress.py

@@ -24,6 +24,7 @@
 from lightning.pytorch.callbacks.progress.progress_bar import ProgressBar
 from lightning.pytorch.utilities.imports import _RICH_AVAILABLE
 from lightning.pytorch.utilities.types import STEP_OUTPUT
+from lightning_utilities.core.apply_func import apply_to_collection
 
 if _RICH_AVAILABLE:
     from rich import get_console, reconfigure