[Issue #38] Handle ModuleNotFoundError for word_levenshtein

Author: wq2012

DiarizationLM/diarizationlm/metrics.py

@@ -28,7 +28,10 @@
 from scipy import optimize
 import tqdm
 from diarizationlm import utils
-import word_levenshtein as levenshtein
+try:
+  import word_levenshtein as levenshtein
+except ModuleNotFoundError:
+  import levenshtein
 
 
 @dataclasses.dataclass

DiarizationLM/diarizationlm/utils.py

@@ -9,7 +9,10 @@
 import numpy as np
 from scipy import optimize
 
-import word_levenshtein as levenshtein
+try:
+  import word_levenshtein as levenshtein
+except ModuleNotFoundError:
+  import levenshtein
 
 PUNCTUATIONS = [",", ".", "_", "?", "!", "-", '"', "'"]