[wheel] disable global logging config (#241)

wenet-e2e · Jun 6, 2024 · d9f47ca · d9f47ca
1 parent c559848
commit d9f47ca
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/tn/processor.py b/tn/processor.py
@@ -23,9 +23,6 @@
 from pynini.lib import byte, utf8
 from pynini.lib.pynutil import delete, insert
 
-logging.basicConfig(level=logging.DEBUG,
-                    format='%(asctime)s WETEXT %(levelname)s %(message)s')
-
 
 class Processor:
 
@@ -78,6 +75,13 @@ def build_verbalizer(self):
         self.verbalizer = self.delete_tokens(verbalizer)
 
     def build_fst(self, prefix, cache_dir, overwrite_cache):
+        logger = logging.getLogger('wetext-{}'.format(self.name))
+        logger.setLevel(logging.INFO)
+        handler = logging.StreamHandler()
+        fmt = logging.Formatter('%(asctime)s WETEXT %(levelname)s %(message)s')
+        handler.setFormatter(fmt)
+        logger.addHandler(handler)
+
         os.makedirs(cache_dir, exist_ok=True)
         tagger_name = '{}_tagger.fst'.format(prefix)
         verbalizer_name = '{}_verbalizer.fst'.format(prefix)
@@ -88,20 +92,20 @@ def build_fst(self, prefix, cache_dir, overwrite_cache):
         exists = os.path.exists(tagger_path) and os.path.exists(
             verbalizer_path)
         if exists and not overwrite_cache:
-            logging.info("found existing fst: {}".format(tagger_path))
-            logging.info("                    {}".format(verbalizer_path))
-            logging.info("skip building fst for {} ...".format(self.name))
+            logger.info("found existing fst: {}".format(tagger_path))
+            logger.info("                    {}".format(verbalizer_path))
+            logger.info("skip building fst for {} ...".format(self.name))
             self.tagger = Fst.read(tagger_path).optimize()
             self.verbalizer = Fst.read(verbalizer_path).optimize()
         else:
-            logging.info("building fst for {} ...".format(self.name))
+            logger.info("building fst for {} ...".format(self.name))
             self.build_tagger()
             self.build_verbalizer()
             self.tagger.optimize().write(tagger_path)
             self.verbalizer.optimize().write(verbalizer_path)
-            logging.info("done")
-            logging.info("fst path: {}".format(tagger_path))
-            logging.info("          {}".format(verbalizer_path))
+            logger.info("done")
+            logger.info("fst path: {}".format(tagger_path))
+            logger.info("          {}".format(verbalizer_path))
 
     def tag(self, input):
         if len(input) == 0: