add mode for tihu

b00f · Sep 28, 2019 · accb2be · accb2be
1 parent b8816ab
commit accb2be
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/lilak.py b/src/lilak.py
@@ -94,7 +94,7 @@ def __init__(self, mode = 0):
 
 
     def is_kam_dandane(self, word):
-        if self.mode == 1:
+        if self.mode == 'tihu':
             return True
 
         dandane = 0
@@ -730,7 +730,7 @@ def pars_main_dic(self):
     parser.add_argument("-o", "--output", help="input dictionary file")
     args = parser.parse_args()
 
-    if args.mode:
+    if args.mode == 'tihu':
         lilak = Lilak(args.mode)
         lilak.read_lexicon(args.input)
         lilak.pars_main_dic()