[Small] Fixes logging (activeloopai#1153)

gaurav879 · Sep 13, 2021 · 2e0fa1d · 2e0fa1d
1 parent 6588072
commit 2e0fa1d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/conftest.py b/conftest.py
@@ -7,7 +7,7 @@
 os.environ["BUGGER_OFF"] = "true"
 
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
-logging.basicConfig(level=logging.ERROR)
+logging.disable(logging.INFO)
 
 from hub.constants import *
 from hub.tests.common import SESSION_ID

diff --git a/hub/client/log.py b/hub/client/log.py
@@ -8,3 +8,11 @@
 import sys
 
 logger = logging.getLogger("hub")
+
+
+def configure_logger(debug=0):
+    log_level = logging.DEBUG if debug == 1 else logging.INFO
+    logging.basicConfig(format="%(message)s", level=log_level, stream=sys.stdout)
+
+
+configure_logger(0)