Skip to content

Commit

Permalink
isort and black
Browse files Browse the repository at this point in the history
  • Loading branch information
lezwon authored and Borda committed Oct 2, 2020
1 parent 1402f1a commit 0a1a435
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion pytorch_lightning/utilities/xla_device_utils.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
import functools
import importlib
import torch
from multiprocessing import Process, Queue

import torch

TORCHXLA_AVAILABLE = importlib.util.find_spec("torch_xla") is not None
if TORCHXLA_AVAILABLE:
import torch_xla.core.xla_model as xm
Expand All @@ -15,6 +16,7 @@ def inner_f(queue, func, **kwargs): # pragma: no cover
queue.put(func(**kwargs))
except Exception as _e:
import traceback

traceback.print_exc()
queue.put(None)

Expand Down

0 comments on commit 0a1a435

Please sign in to comment.