Merge branch 'dep_mon' of github.com:daniellepintz/pytorch-lightning into dep_mon

Author: daniellepintz

pytorch_lightning/callbacks/gpu_stats_monitor.py

@@ -29,7 +29,7 @@
 
 import pytorch_lightning as pl
 from pytorch_lightning.callbacks.base import Callback
-from pytorch_lightning.utilities import DeviceType, rank_zero_only, rank_zero_deprecation
+from pytorch_lightning.utilities import DeviceType, rank_zero_deprecation, rank_zero_only
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 from pytorch_lightning.utilities.parsing import AttributeDict
 from pytorch_lightning.utilities.types import STEP_OUTPUT

pytorch_lightning/callbacks/xla_stats_monitor.py

@@ -21,7 +21,7 @@
 import time
 
 from pytorch_lightning.callbacks.base import Callback
-from pytorch_lightning.utilities import _TPU_AVAILABLE, DeviceType, rank_zero_info, rank_zero_deprecation
+from pytorch_lightning.utilities import _TPU_AVAILABLE, DeviceType, rank_zero_deprecation, rank_zero_info
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 
 if _TPU_AVAILABLE:

tests/deprecated_api/test_remove_1-7.py

@@ -18,8 +18,7 @@
 import torch
 
 from pytorch_lightning import Callback, LightningDataModule, Trainer
-from pytorch_lightning.callbacks.xla_stats_monitor import GPUStatsMonitor
-from pytorch_lightning.callbacks.xla_stats_monitor import XLAStatsMonitor
+from pytorch_lightning.callbacks.xla_stats_monitor import GPUStatsMonitor, XLAStatsMonitor
 from pytorch_lightning.loggers import LoggerCollection, TestTubeLogger
 from tests.deprecated_api import _soft_unimport_module
 from tests.helpers import BoringModel