diff --git a/CHANGELOG.md b/CHANGELOG.md index 482525fd3542a..a3061cdada105 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -117,6 +117,9 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/). - Removed duplicated file extension when uploading model checkpoints with `NeptuneLogger` ([#11015](https://github.com/PyTorchLightning/pytorch-lightning/pull/11015)) +- Removed `__getstate__` and `__setstate__` of `RichProgressBar` ([#11100](https://github.com/PyTorchLightning/pytorch-lightning/pull/11100)) + + - The `DDPPlugin` and `DDPSpawnPlugin` and their subclasses now remove the `SyncBatchNorm` wrappers in `teardown()` to enable proper support at inference after fitting ([#11078](https://github.com/PyTorchLightning/pytorch-lightning/pull/11078)) diff --git a/pytorch_lightning/callbacks/progress/rich_progress.py b/pytorch_lightning/callbacks/progress/rich_progress.py index 2e24df6da61f6..f983ccdaab12e 100644 --- a/pytorch_lightning/callbacks/progress/rich_progress.py +++ b/pytorch_lightning/callbacks/progress/rich_progress.py @@ -318,17 +318,6 @@ def on_validation_start(self, trainer, pl_module): super().on_validation_start(trainer, pl_module) self._init_progress(trainer) - def __getstate__(self): - # can't pickle the rich progress objects - state = self.__dict__.copy() - state["progress"] = None - state["_console"] = None - return state - - def __setstate__(self, state): - self.__dict__ = state - self._console = Console(**self._console_kwargs) - def on_sanity_check_start(self, trainer, pl_module): super().on_sanity_check_start(trainer, pl_module) self._init_progress(trainer)