update gpt corrector.

examples/gpt/demo.py

@@ -6,7 +6,7 @@
 import sys
 
 sys.path.append("../..")
-from pycorrector import GptCorrector
+from pycorrector.gpt.gpt_corrector import GptCorrector
 
 if __name__ == '__main__':
     error_sentences = [

pycorrector/__init__.py

@@ -11,7 +11,6 @@ from pycorrector.detector import Detector
 from pycorrector.detector import USER_DATA_DIR
 from pycorrector.en_spell_corrector import EnSpellCorrector
 from pycorrector.ernie_csc.ernie_csc_corrector import ErnieCscCorrector
-from pycorrector.gpt.gpt_corrector import GptCorrector
 from pycorrector.macbert.macbert_corrector import MacBertCorrector
 from pycorrector.proper_corrector import ProperCorrector
 from pycorrector.seq2seq.conv_seq2seq_corrector import ConvSeq2SeqCorrector

pycorrector/confusion_corrector.py

@@ -7,7 +7,6 @@
 import os
 from typing import List
 
-from ahocorasick import Automaton
 from loguru import logger
 
 
@@ -21,7 +20,11 @@ class ConfusionCorrector:
             self.custom_confusion = self.load_custom_confusion_dict(custom_confusion_path_or_dict)
         else:
             raise ValueError('custom_confusion_path_or_dict must be dict or str.')
-
+        try:
+            from ahocorasick import Automaton
+        except ImportError:
+            logger.error('ahocorasick package not found. pip install pyahocorasick')
+            raise ImportError('ahocorasick package not found. pip install pyahocorasick')
         self.automaton = Automaton()
         for idx, (err, truth) in enumerate(self.custom_confusion.items()):
             self.automaton.add_word(err, (idx, err, truth))

pycorrector/version.py

@@ -1 +1 @@
-__version__ = '1.0.5'
+__version__ = '1.0.6'