Skip to content

Commit 39a557c

Browse files
committed
Update data_helpers.py
1 parent 588a7e0 commit 39a557c

1 file changed

Lines changed: 13 additions & 5 deletions

File tree

‎utils/data_helpers.py‎

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -50,16 +50,24 @@ def logger_fn(name, input_file, level=logging.INFO):
5050
Returns:
5151
The logger
5252
"""
53-
tf_logger = logging.getLogger(name)
54-
tf_logger.setLevel(level)
53+
logger = logging.getLogger(name)
54+
logger.setLevel(level)
5555
log_dir = os.path.dirname(input_file)
5656
if not os.path.exists(log_dir):
5757
os.makedirs(log_dir)
58-
fh = logging.FileHandler(input_file, mode='w')
5958
formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
59+
60+
# File Handler
61+
fh = logging.FileHandler(input_file, mode='w')
6062
fh.setFormatter(formatter)
61-
tf_logger.addHandler(fh)
62-
return tf_logger
63+
logger.addHandler(fh)
64+
65+
# stream Handler
66+
sh = logging.StreamHandler()
67+
sh.setFormatter(formatter)
68+
sh.setLevel(logging.INFO)
69+
logger.addHandler(sh)
70+
return logger
6371

6472

6573
def tab_printer(args, logger):

0 commit comments

Comments
 (0)