flake8

Lightning-AI · Mar 29, 2021 · d178e1d · d178e1d
1 parent 4509771
commit d178e1d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pytorch_lightning/utilities/xla_device.py b/pytorch_lightning/utilities/xla_device.py
@@ -16,7 +16,6 @@
 import queue as q
 import traceback
 from multiprocessing import Process, Queue
-from typing import Optional
 
 import torch.multiprocessing as mp
 
@@ -71,7 +70,7 @@ def _is_device_tpu() -> bool:
             A boolean value indicating if the xla device is a TPU device or not
         """
 
-        def _fn(process_idx: int, mp_queue):
+        def _fn(_: int, mp_queue):
             try:
                 device = xm.xla_device()
                 mp_queue.put(device.type == 'xla')