fix import

google · Sep 15, 2024 · 6cf7798 · 6cf7798
1 parent 3a8f092
commit 6cf7798
Show file tree

Hide file tree

Showing 9 changed files with 20 additions and 9 deletions.
diff --git a/lingvo/setup.py b/lingvo/setup.py
@@ -2,7 +2,7 @@
 
 import setuptools
 
-VERSION = "0.0.7"
+VERSION = "0.0.8"
 
 with open("README.md", "r") as file_object:
     LONG_DESCRIPTION = file_object.read()

diff --git a/lingvo/sidlingvo/__init__.py b/lingvo/sidlingvo/__init__.py
@@ -0,0 +1,11 @@
+"""__init__ file."""
+
+from . import fe_utils
+from . import wav_to_dvector
+from . import wav_to_lang
+
+load_tflite_model = fe_utils.load_tflite_model
+
+WavToLangRunner = wav_to_lang.WavToLangRunner
+
+WavToDvectorRunner = wav_to_dvector.WavToDvectorRunner
diff --git a/lingvo/sidlingvo/attentive_scoring_layer_test.py b/lingvo/sidlingvo/attentive_scoring_layer_test.py
@@ -1,6 +1,6 @@
 """Tests for attentive scoring."""
 
-import attentive_scoring_layer
+from sidlingvo import attentive_scoring_layer
 from lingvo import compat as tf
 from lingvo.core import test_utils
 import numpy as np

diff --git a/lingvo/sidlingvo/cumulative_statistics_layer_test.py b/lingvo/sidlingvo/cumulative_statistics_layer_test.py
@@ -6,7 +6,7 @@
 from lingvo.core import test_utils
 import numpy as np
 
-import cumulative_statistics_layer
+from sidlingvo import cumulative_statistics_layer
 
 _TF_RANDOM_SEED = 1314
 

diff --git a/lingvo/sidlingvo/fe_utils_test.py b/lingvo/sidlingvo/fe_utils_test.py
@@ -2,7 +2,7 @@
 from lingvo.core import test_utils
 import numpy as np
 
-import fe_utils
+from sidlingvo import fe_utils
 
 
 class FrontendTest(test_utils.TestCase):

diff --git a/lingvo/sidlingvo/loss_layers.py b/lingvo/sidlingvo/loss_layers.py
@@ -6,7 +6,7 @@
 from lingvo.core import layers as lingvo_layers
 from lingvo.core import py_utils
 import attentive_scoring_layer
-import utils
+from sidlingvo import utils
 
 
 class EmbeddingComparisonType(enum.Enum):

diff --git a/lingvo/sidlingvo/loss_layers_test.py b/lingvo/sidlingvo/loss_layers_test.py
@@ -5,7 +5,7 @@
 from lingvo.core import test_utils
 
 import numpy as np
-import loss_layers
+from sidlingvo import loss_layers
 
 FLAGS = tf.flags.FLAGS
 _RANDOM_SEED = 1314

diff --git a/lingvo/sidlingvo/utils_test.py b/lingvo/sidlingvo/utils_test.py
@@ -5,7 +5,7 @@
 
 import numpy as np
 
-import utils
+from sidlingvo import utils
 
 _RANDOM_SEED = 1321
 

diff --git a/lingvo/sidlingvo/wav_to_lang.py b/lingvo/sidlingvo/wav_to_lang.py
@@ -5,8 +5,8 @@
 from lingvo.core import py_utils
 import numpy as np
 
-import fe_utils
-import language_map
+from sidlingvo import fe_utils
+from sidlingvo import language_map
 
 
 @dataclasses.dataclass