From 1f60032e96b6407012f83c47ac0dfc165e540676 Mon Sep 17 00:00:00 2001
From: David Eliahu <deliahu@users.noreply.github.com>
Date: Thu, 8 Aug 2019 21:51:24 -0700
Subject: [PATCH] Update TF_CPP_MIN_LOG_LEVEL

(cherry picked from commit dcf8ed9d5760f5bd166d714e1eef47bc7ce2e802)
---
 pkg/workloads/cortex/lib/tf_lib.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/pkg/workloads/cortex/lib/tf_lib.py b/pkg/workloads/cortex/lib/tf_lib.py
index f325ee21c5..c80ef5b19e 100644
--- a/pkg/workloads/cortex/lib/tf_lib.py
+++ b/pkg/workloads/cortex/lib/tf_lib.py
@@ -31,9 +31,12 @@
 }
 
 
+CPP_LOG_LEVEL = {"DEBUG": "1", "INFO": "2", "WARN": "3", "ERROR": "4", "FATAL": "5"}
+
+
 def set_logging_verbosity(verbosity):
     tf.logging.set_verbosity(verbosity)
-    os.environ["TF_CPP_MIN_LOG_LEVEL"] = str(tf.logging.__dict__[verbosity] / 10)
+    os.environ["TF_CPP_MIN_LOG_LEVEL"] = CPP_LOG_LEVEL[verbosity]
 
 
 def get_column_tf_types(model_name, ctx, training=True):