added todo

Lightning-AI · Sep 30, 2020 · 051df9b · 051df9b
1 parent 28bd155
commit 051df9b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pytorch_lightning/accelerators/tpu_backend.py b/pytorch_lightning/accelerators/tpu_backend.py
@@ -46,6 +46,7 @@ def __init__(self, trainer):
     def setup(self, model):
         rank_zero_info(f'training on {self.trainer.tpu_cores} TPU cores')
 
+        # TODO: Move this check to Trainer __init__ or device parser
         if not TPU_AVAILABLE:
             raise MisconfigurationException('PyTorch XLA not installed.')