[Tn&ITN] Fix issue #96 (#97)

Author: xingchensong

itn/chinese/inverse_normalizer.py

@@ -30,13 +30,15 @@
 
 class InverseNormalizer(Processor):
 
-    def __init__(self, cache_dir='itn', overwrite_cache=False,
+    def __init__(self, cache_dir=None, overwrite_cache=False,
                  enable_standalone_number=True,
                  enable_0_to_9=True):
         super().__init__(name='inverse_normalizer', ordertype='itn')
         self.convert_number = enable_standalone_number
         self.enable_0_to_9 = enable_0_to_9
-        self.build_fst('zh_itn', files(cache_dir), overwrite_cache)
+        if cache_dir is None:
+            cache_dir = files("itn")
+        self.build_fst('zh_itn', cache_dir, overwrite_cache)
 
     def build_tagger(self):
         tagger = (add_weight(Date().tagger, 1.02)

tn/chinese/normalizer.py

@@ -33,7 +33,7 @@
 class Normalizer(Processor):
 
     def __init__(self,
-                 cache_dir='tn',
+                 cache_dir=None,
                  overwrite_cache=False,
                  remove_interjections=True,
                  traditional_to_simple=True,
@@ -46,7 +46,9 @@ def __init__(self,
         self.remove_puncts = remove_puncts
         self.full_to_half = full_to_half
         self.tag_oov = tag_oov
-        self.build_fst('zh_tn', files(cache_dir), overwrite_cache)
+        if cache_dir is None:
+            cache_dir = files("tn")
+        self.build_fst('zh_tn', cache_dir, overwrite_cache)
 
     def build_tagger(self):
         processor = PreProcessor(