Skip to content

Commit 92b1d69

Browse files
committed
fix mypy and device string format.
1 parent 1d9bd0d commit 92b1d69

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lightning/pytorch/trainer/setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ def _init_profiler(trainer: "pl.Trainer", profiler: Optional[Union[Profiler, str
144144
def _log_device_info(trainer: "pl.Trainer") -> None:
145145
if CUDAAccelerator.is_available():
146146
if isinstance(trainer.accelerator, CUDAAccelerator):
147-
device_name = list({CUDAAccelerator.device_name(d) for d in trainer.device_ids})
147+
device_name = ", ".join(list({CUDAAccelerator.device_name(d) for d in trainer.device_ids}))
148148
else:
149149
device_name = CUDAAccelerator.device_name()
150150
elif MPSAccelerator.is_available():

0 commit comments

Comments
 (0)