wit: another tf2 compat change

Author: stephanwlee

tensorboard/plugins/interactive_inference/witwidget/notebook/base.py

@@ -36,7 +36,7 @@ def __init__(self, config_builder):
     Args:
       config_builder: WitConfigBuilder object containing settings for WIT.
     """
-    tf.logging.set_verbosity(tf.logging.WARN)
+    tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.WARN)
     config = config_builder.build()
     copied_config = dict(config)
     self.estimator_and_spec = (