Skip to content

Commit 88e65e8

Browse files
committed
Update CHANGELOG.md
1 parent 036ecf2 commit 88e65e8

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

CHANGELOG.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,9 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
9898
- Added a `_Stateful` support for `LightningDataModule` ([#11637](https://github.com/PyTorchLightning/pytorch-lightning/pull/11637))
9999

100100

101+
- Added checks to `TPUAccelerator.setup_environment` to assert device availability ([#11799](https://github.com/PyTorchLightning/pytorch-lightning/pull/11799))
102+
103+
101104
### Changed
102105

103106
- Implemented a new native and rich format in `_print_results` method of the `EvaluationLoop` ([#11332](https://github.com/PyTorchLightning/pytorch-lightning/pull/11332))

pytorch_lightning/accelerators/tpu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
class TPUAccelerator(Accelerator):
2727
"""Accelerator for TPU devices."""
2828

29-
def setup_environment(self, root_device: torch.device) -> None:
29+
def __init__(self) -> None:
3030
"""
3131
Raises:
3232
MisconfigurationException:

0 commit comments

Comments
 (0)