diff --git a/src/lightning/pytorch/loops/progress.py b/src/lightning/pytorch/loops/progress.py index d2e52f44d7ba7f..367ae79530ed53 100644 --- a/src/lightning/pytorch/loops/progress.py +++ b/src/lightning/pytorch/loops/progress.py @@ -124,7 +124,7 @@ class _Progress(_BaseProgress): current: _ReadyCompletedTracker = field(default_factory=_ProcessedTracker) def __post_init__(self) -> None: - if type(self.total) is not type(self.current): # noqa: E721 + if type(self.total) is not type(self.current): raise ValueError("The `total` and `current` instances should be of the same class") def increment_ready(self) -> None: