remove dependency for `python-Levenshtein`

cnocr/cli.py

@@ -28,7 +28,6 @@ from operator import itemgetter
 from pathlib import Path
 
 import click
-import Levenshtein
 from torchvision import transforms as T
 import torch
 
@@ -275,6 +274,12 @@ def evaluate(
     verbose,
 ):
     """评估模型效果"""
+    try:
+        import Levenshtein
+    except Exception as e:
+        logger.error(e)
+        logger.error('try to install the package by using `pip install python-Levenshtein`')
+        return
     ocr = CnOcr(model_name=model_name, model_fp=pretrained_model_fp, context=context)
 
     fn_labels_list = read_input_file(eval_index_fp)

tests/test_cnocr.py

@@ -25,7 +25,6 @@ import pytest
 
 import numpy as np
 from PIL import Image
-import Levenshtein
 
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 sys.path.insert(1, os.path.dirname(os.path.abspath(__file__)))
@@ -99,6 +98,8 @@ def print_preds(pred):
 
 
 def cal_score(preds, expected):
+    import Levenshtein
+
     if len(preds) != len(expected):
         return 0
     total_cnt = 0