Skip to content

Commit

Permalink
[Minor] set to staticmethod
Browse files Browse the repository at this point in the history
  • Loading branch information
fedebotu committed Apr 28, 2024
1 parent 5e0e02b commit 97c83e3
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -327,7 +327,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."""
if XLAAccelerator.is_available():
return "tpu"
Expand Down

0 comments on commit 97c83e3

Please sign in to comment.