Fix pytype

google · Oct 21, 2024 · 6e061a9 · 6e061a9
1 parent 98ad96e
commit 6e061a9
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/DiarizationLM/diarizationlm/__init__.py b/DiarizationLM/diarizationlm/__init__.py
@@ -1,8 +1,10 @@
 """__init__ file."""
 
+from . import levenshtein
 from . import utils
 from . import metrics
 
+levenshtein_with_edits = levenshtein.levenshtein_with_edits
 PromptOptions = utils.PromptOptions
 transcript_preserving_speaker_transfer = (
     utils.transcript_preserving_speaker_transfer)

diff --git a/DiarizationLM/diarizationlm/metrics.py b/DiarizationLM/diarizationlm/metrics.py
@@ -31,7 +31,7 @@
 try:
   import word_levenshtein as levenshtein
 except ModuleNotFoundError:
-  import levenshtein
+  from diarizationlm import levenshtein
 
 
 @dataclasses.dataclass

diff --git a/DiarizationLM/diarizationlm/utils.py b/DiarizationLM/diarizationlm/utils.py
@@ -12,7 +12,7 @@
 try:
   import word_levenshtein as levenshtein
 except ModuleNotFoundError:
-  import levenshtein
+  from diarizationlm import levenshtein
 
 PUNCTUATIONS = [",", ".", "_", "?", "!", "-", '"', "'"]
 

diff --git a/DiarizationLM/setup.py b/DiarizationLM/setup.py
@@ -2,7 +2,7 @@
 
 import setuptools
 
-VERSION = "0.1.2"
+VERSION = "0.1.3"
 
 with open("README.md", "r") as file_object:
     LONG_DESCRIPTION = file_object.read()