diff --git a/src/dvclive/fabric.py b/src/dvclive/fabric.py index 0e0903d..cfda948 100644 --- a/src/dvclive/fabric.py +++ b/src/dvclive/fabric.py @@ -11,8 +11,8 @@ ) from lightning.fabric.utilities.rank_zero import rank_zero_only except ImportError: - from lightning_fabric.loggers.logger import Logger, rank_zero_experiment - from lightning_fabric.utilities.logger import ( + from lightning_fabric.loggers.logger import Logger, rank_zero_experiment # type: ignore[assignment] + from lightning_fabric.utilities.logger import ( # type: ignore[assignment] _add_prefix, _convert_params, _sanitize_callable_params,