Fix registry descriptions (#18766)

src/lightning/fabric/accelerators/cpu.py

@@ -60,7 +60,7 @@ def register_accelerators(cls, accelerator_registry: _AcceleratorRegistry) -> No
         accelerator_registry.register(
             "cpu",
             cls,
-            description=cls.__class__.__name__,
+            description=cls.__name__,
         )
 
 

src/lightning/fabric/accelerators/cuda.py

@@ -68,7 +68,7 @@ def register_accelerators(cls, accelerator_registry: _AcceleratorRegistry) -> No
         accelerator_registry.register(
             "cuda",
             cls,
-            description=cls.__class__.__name__,
+            description=cls.__name__,
         )
 
 

src/lightning/fabric/accelerators/mps.py

@@ -70,7 +70,7 @@ def register_accelerators(cls, accelerator_registry: _AcceleratorRegistry) -> No
         accelerator_registry.register(
             "mps",
             cls,
-            description=cls.__class__.__name__,
+            description=cls.__name__,
         )
 
 

src/lightning/fabric/accelerators/xla.py

@@ -95,7 +95,7 @@ def is_available() -> bool:
 
     @classmethod
     def register_accelerators(cls, accelerator_registry: _AcceleratorRegistry) -> None:
-        accelerator_registry.register("tpu", cls, description=cls.__class__.__name__)
+        accelerator_registry.register("tpu", cls, description=cls.__name__)
 
 
 # PJRT support requires this minimum version

src/lightning/fabric/strategies/dp.py

@@ -98,4 +98,4 @@ def load_module_state_dict(
 
     @classmethod
     def register_strategies(cls, strategy_registry: _StrategyRegistry) -> None:
-        strategy_registry.register("dp", cls, description=cls.__class__.__name__)
+        strategy_registry.register("dp", cls, description=cls.__name__)

src/lightning/fabric/strategies/single_xla.py

@@ -62,4 +62,4 @@ def checkpoint_io(self, io: CheckpointIO) -> None:
 
     @classmethod
     def register_strategies(cls, strategy_registry: _StrategyRegistry) -> None:
-        strategy_registry.register("single_xla", cls, description=cls.__class__.__name__)
+        strategy_registry.register("single_xla", cls, description=cls.__name__)

src/lightning/fabric/strategies/xla.py

@@ -264,4 +264,4 @@ def save_checkpoint(
 
     @classmethod
     def register_strategies(cls, strategy_registry: _StrategyRegistry) -> None:
-        strategy_registry.register("xla", cls, description=cls.__class__.__name__)
+        strategy_registry.register("xla", cls, description=cls.__name__)

src/lightning/fabric/strategies/xla_fsdp.py

@@ -594,7 +594,7 @@ def load_checkpoint(
 
     @classmethod
     def register_strategies(cls, strategy_registry: _StrategyRegistry) -> None:
-        strategy_registry.register("xla_fsdp", cls, description=cls.__class__.__name__)
+        strategy_registry.register("xla_fsdp", cls, description=cls.__name__)
 
     def _parse_fsdp_kwargs(self) -> Dict:
         # this needs to be delayed because `self.precision` isn't available at init

src/lightning/pytorch/accelerators/cpu.py

@@ -68,7 +68,7 @@ def register_accelerators(cls, accelerator_registry: _AcceleratorRegistry) -> No
         accelerator_registry.register(
             "cpu",
             cls,
-            description=cls.__class__.__name__,
+            description=cls.__name__,
         )
 
 

src/lightning/pytorch/accelerators/cuda.py

@@ -100,7 +100,7 @@ def register_accelerators(cls, accelerator_registry: _AcceleratorRegistry) -> No
         accelerator_registry.register(
             "cuda",
             cls,
-            description=f"{cls.__class__.__name__}",
+            description=cls.__name__,
         )
 
 

src/lightning/pytorch/accelerators/mps.py

@@ -75,7 +75,7 @@ def register_accelerators(cls, accelerator_registry: _AcceleratorRegistry) -> No
         accelerator_registry.register(
             "mps",
             cls,
-            description=cls.__class__.__name__,
+            description=cls.__name__,
         )
 
 

src/lightning/pytorch/accelerators/xla.py

@@ -48,4 +48,4 @@ def get_device_stats(self, device: _DEVICE) -> Dict[str, Any]:
 
     @classmethod
     def register_accelerators(cls, accelerator_registry: _AcceleratorRegistry) -> None:
-        accelerator_registry.register("tpu", cls, description=cls.__class__.__name__)
+        accelerator_registry.register("tpu", cls, description=cls.__name__)

src/lightning/pytorch/strategies/single_device.py

@@ -92,5 +92,5 @@ def register_strategies(cls, strategy_registry: _StrategyRegistry) -> None:
         strategy_registry.register(
             cls.strategy_name,
             cls,
-            description=cls.__class__.__name__,
+            description=cls.__name__,
         )

src/lightning/pytorch/strategies/single_xla.py

@@ -78,7 +78,7 @@ def setup(self, trainer: "pl.Trainer") -> None:
 
     @classmethod
     def register_strategies(cls, strategy_registry: _StrategyRegistry) -> None:
-        strategy_registry.register("single_xla", cls, description=cls.__class__.__name__)
+        strategy_registry.register("single_xla", cls, description=cls.__name__)
 
     def teardown(self) -> None:
         super().teardown()

src/lightning/pytorch/strategies/xla.py

@@ -312,7 +312,7 @@ def register_strategies(cls, strategy_registry: _StrategyRegistry) -> None:
         strategy_registry.register(
             cls.strategy_name,
             cls,
-            description=f"{cls.__class__.__name__}",
+            description=cls.__name__,
         )
 
     def _pod_progress_bar_force_stdout(self) -> None: