Fix test

Lightning-AI · Mar 14, 2020 · f45b02a · f45b02a
1 parent e7e9f1f
commit f45b02a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/pytorch_lightning/core/memory.py b/pytorch_lightning/core/memory.py
@@ -10,7 +10,7 @@
 import numpy as np
 import torch
 
-from ..core import root_logger as log
+from pytorch_lightning.core import root_logger as log
 
 class ModelSummary(object):
 

diff --git a/pytorch_lightning/trainer/training_io.py b/pytorch_lightning/trainer/training_io.py
@@ -89,7 +89,6 @@
 
 """
 
-import logging as log
 import os
 import re
 import signal