From 68f9a41f75e7f87f4b33102e76407b712934c435 Mon Sep 17 00:00:00 2001
From: Ryan Wolbeck <wolbrya@chrobinson.com>
Date: Fri, 14 Aug 2020 14:52:19 -0500
Subject: [PATCH] update the default for compression and add some better
 logging

---
 ngboost_tuner/cli.py  | 2 +-
 ngboost_tuner/tune.py | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/ngboost_tuner/cli.py b/ngboost_tuner/cli.py
index 5682f4d..e37b66b 100644
--- a/ngboost_tuner/cli.py
+++ b/ngboost_tuner/cli.py
@@ -54,7 +54,7 @@ def build_cli():
         "-ct",
         "--compression-type",
         type=str,
-        default=os.getenv("COMPRESSION"),
+        default=os.getenv("COMPRESSION", None),
         help="Input data compression, i.e. gzip or None; defaults to $COMPRESSION if not set",
     )
     tune.add_argument(
diff --git a/ngboost_tuner/tune.py b/ngboost_tuner/tune.py
index 13771eb..ce29ce0 100644
--- a/ngboost_tuner/tune.py
+++ b/ngboost_tuner/tune.py
@@ -233,6 +233,8 @@ def objective(params):
             Y_val=y_valid.values,
             early_stopping_rounds=2,
         )
+        log.info(f"Final LightGBM parameters {final_params_lightgbm}")
+        log.info(f"Final NGBoost parameters {final_ngboost_params}")
 
     else:
         base_models = [