set logger level for package (#1718)

* move logging config to trainer class init * alternate logging config
Lightning-AI · May 12, 2020 · 1df0d2d · 1df0d2d
1 parent 4b30ef6
commit 1df0d2d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pytorch_lightning/__init__.py b/pytorch_lightning/__init__.py
@@ -34,7 +34,8 @@
 import logging as python_logging
 
 _logger = python_logging.getLogger("lightning")
-python_logging.basicConfig(level=python_logging.INFO)
+_logger.addHandler(python_logging.StreamHandler())
+_logger.setLevel(python_logging.INFO)
 
 try:
     # This variable is injected in the __builtins__ by the build

diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
@@ -1,5 +1,6 @@
 import inspect
 import os
+import logging as python_logging
 from argparse import ArgumentParser
 from typing import Union, Optional, List, Dict, Tuple, Iterable, Any