diff --git a/lightgbmlss/logger.py b/lightgbmlss/logger.py index aa1ff04..4571885 100644 --- a/lightgbmlss/logger.py +++ b/lightgbmlss/logger.py @@ -5,7 +5,7 @@ class CustomLogger: def __init__(self): self.logger = logging.getLogger('lightgbm_custom') self.logger.setLevel(logging.ERROR) - + def info(self, message): self.logger.info(message)