Skip to content

Commit 8cb809f

Browse files
committed
add override decorator.
1 parent 3a3949a commit 8cb809f

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

src/lightning/pytorch/accelerators/cuda.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,7 @@ def register_accelerators(cls, accelerator_registry: _AcceleratorRegistry) -> No
114114
)
115115

116116
@classmethod
117+
@override
117118
def device_name(cls, device: Optional[_DEVICE] = None) -> str:
118119
if not cls.is_available():
119120
return "False"

src/lightning/pytorch/accelerators/xla.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@ def register_accelerators(cls, accelerator_registry: _AcceleratorRegistry) -> No
5656
accelerator_registry.register("tpu", cls, description=cls.__name__)
5757

5858
@classmethod
59+
@override
5960
def device_name(cls, device: Optional[_DEVICE] = None) -> str:
6061
is_available = cls.is_available()
6162
if not is_available:

0 commit comments

Comments
 (0)