From f90a79a86864e07913df12200d26200e44e013e3 Mon Sep 17 00:00:00 2001 From: awaelchli Date: Wed, 5 Jun 2024 02:04:22 +0200 Subject: [PATCH] also for Fabric --- src/lightning/fabric/connector.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/lightning/fabric/connector.py b/src/lightning/fabric/connector.py index edbfd77721a95..3e1cd8cc46790 100644 --- a/src/lightning/fabric/connector.py +++ b/src/lightning/fabric/connector.py @@ -312,7 +312,8 @@ def _check_device_config_and_set_final_flags(self, devices: Union[List[int], str f" using {accelerator_name} accelerator." ) - def _choose_auto_accelerator(self) -> str: + @staticmethod + def _choose_auto_accelerator() -> str: """Choose the accelerator type (str) based on availability when ``accelerator='auto'``.""" if XLAAccelerator.is_available(): return "tpu"