From 1dfb6da4f8f1fec01d82f5f2dcbc66751d4078ec Mon Sep 17 00:00:00 2001
From: Birte Kristina Friesel <birte.friesel@uos.de>
Date: Thu, 7 Mar 2024 09:34:32 +0100
Subject: lightgbm: use same logger as dfatool

---
 lib/functions.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'lib')

diff --git a/lib/functions.py b/lib/functions.py
index 417c8c8..efed2f6 100644
--- a/lib/functions.py
+++ b/lib/functions.py
@@ -1054,6 +1054,7 @@ class LightGBMFunction(SKLearnRegressionFunction):
 
         import dfatool.lightgbm as lightgbm
 
+        lightgbm.register_logger(logger)
         lgbr = lightgbm.LGBMRegressor(
             boosting_type=boosting_type,
             n_estimators=n_estimators,
-- 
cgit v1.2.3