From d11c163909fcf88976416f421f96b0b3596bdebc Mon Sep 17 00:00:00 2001 From: sammer1107 Date: Mon, 6 Mar 2023 17:28:15 +0800 Subject: [PATCH] Conditional import for transformers --- libmultilabel/logging.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/libmultilabel/logging.py b/libmultilabel/logging.py index 9211427c..e87cd937 100644 --- a/libmultilabel/logging.py +++ b/libmultilabel/logging.py @@ -1,5 +1,4 @@ import logging -import transformers.utils.logging as transformer_logging LOG_FORMAT = '%(asctime)s %(levelname)s:%(message)s' @@ -48,8 +47,12 @@ def add_stream_handler(level=logging.INFO): else: logging.getLogger().setLevel(logging.NOTSET) # use handlers to control levels - transformer_logging.disable_default_handler() - transformer_logging.enable_propagation() + try: + import transformers.utils.logging as transformer_logging + transformer_logging.disable_default_handler() + transformer_logging.enable_propagation() + except ImportError: + pass lightning_logger = logging.getLogger('pytorch_lightning') lightning_logger.handlers.clear()