Skip to content

Commit f024a69

Browse files
committed
isort and black
1 parent e9b8c63 commit f024a69

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

pytorch_lightning/utilities/xla_device_utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,9 @@
11
import functools
22
import importlib
3-
import torch
43
from multiprocessing import Process, Queue
54

5+
import torch
6+
67
TORCHXLA_AVAILABLE = importlib.util.find_spec("torch_xla") is not None
78
if TORCHXLA_AVAILABLE:
89
import torch_xla.core.xla_model as xm
@@ -15,6 +16,7 @@ def inner_f(queue, func, **kwargs): # pragma: no cover
1516
queue.put(func(**kwargs))
1617
except Exception as _e:
1718
import traceback
19+
1820
traceback.print_exc()
1921
queue.put(None)
2022

0 commit comments

Comments
 (0)