flake8 & isort (#5647)

pytorch_lightning/core/memory.py

@@ -389,9 +389,11 @@ def get_gpu_memory_map() -> Dict[str, int]:
     }
     return gpu_memory_map
 
+
 def get_formatted_model_size(total_model_size: float) -> float:
     return f"{total_model_size:,.3f}"
 
+
 def get_human_readable_count(number: int) -> str:
     """
     Abbreviates an integer number with K, M, B, T for thousands, millions,

pytorch_lightning/loggers/wandb.py

@@ -29,9 +29,8 @@ from pytorch_lightning.utilities.warning_utils import WarningCache
 _WANDB_AVAILABLE = _module_available("wandb")
 
 try:
-    from wandb.wandb_run import Run
-
     import wandb
+    from wandb.wandb_run import Run
 except ImportError:
     # needed for test mocks, these tests shall be updated
     wandb, Run = None, None

tests/core/test_memory.py

@@ -50,6 +50,7 @@ class PreCalculatedModel(BoringModel):
         x = self.layer(x)
         return self.layer1(x)
 
+
 class UnorderedModel(LightningModule):
     """ A model in which the layers not defined in order of execution """