From 1df0d2dc97e20b9646cbe0f42060a57f99f397fc Mon Sep 17 00:00:00 2001 From: Jeremy Jordan <13970565+jeremyjordan@users.noreply.github.com> Date: Tue, 12 May 2020 00:14:35 -0400 Subject: [PATCH] set logger level for package (#1718) * move logging config to trainer class init * alternate logging config --- pytorch_lightning/__init__.py | 3 ++- pytorch_lightning/trainer/trainer.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/__init__.py b/pytorch_lightning/__init__.py index fb77ead0..f305c06b 100644 --- a/pytorch_lightning/__init__.py +++ b/pytorch_lightning/__init__.py @@ -34,7 +34,8 @@ Documentation 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 index 743d8618..cd3ee1c9 100644 --- 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