[Issue #38] Handle ModuleNotFoundError for word_levenshtein

google · Oct 21, 2024 · f86543d · f86543d
1 parent b115f0f
commit f86543d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/DiarizationLM/diarizationlm/metrics.py b/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

diff --git a/DiarizationLM/diarizationlm/utils.py b/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 = [",", ".", "_", "?", "!", "-", '"', "'"]