diff --git a/auton_survival/__init__.py b/auton_survival/__init__.py
index 0c9b28b..5ad5028 100644
--- a/auton_survival/__init__.py
+++ b/auton_survival/__init__.py
@@ -475,3 +475,6 @@ class is a composite transform that does both Imputing ***and*** Scaling with
 
 import warnings
 from tqdm.auto import tqdm
+import logging
+
+logging.basicConfig()