Hide TF info messages in logs (#492)

ServiceNow · Mar 15, 2023 · 180d3bc · 180d3bc
1 parent 9e89d9e
commit 180d3bc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/azimuth/__init__.py b/azimuth/__init__.py
@@ -1,3 +1,4 @@
+import os
 from pathlib import Path
 
 from datasets import disable_progress_bar
@@ -10,3 +11,4 @@
 set_verbosity_error()
 disable_progress_bar()
 set_logger_config()
+os.environ["TF_CPP_MIN_LOG_LEVEL"] = "1"  # INFO messages from TensorFlow are not printed