From 6cf7798cce85f7ea0f558970c57bb9471566b31f Mon Sep 17 00:00:00 2001
From: Quan Wang <quanw@google.com>
Date: Sat, 14 Sep 2024 23:11:43 -0400
Subject: [PATCH] fix import

---
 lingvo/setup.py                                      |  2 +-
 lingvo/sidlingvo/__init__.py                         | 11 +++++++++++
 lingvo/sidlingvo/attentive_scoring_layer_test.py     |  2 +-
 lingvo/sidlingvo/cumulative_statistics_layer_test.py |  2 +-
 lingvo/sidlingvo/fe_utils_test.py                    |  2 +-
 lingvo/sidlingvo/loss_layers.py                      |  2 +-
 lingvo/sidlingvo/loss_layers_test.py                 |  2 +-
 lingvo/sidlingvo/utils_test.py                       |  2 +-
 lingvo/sidlingvo/wav_to_lang.py                      |  4 ++--
 9 files changed, 20 insertions(+), 9 deletions(-)

diff --git a/lingvo/setup.py b/lingvo/setup.py
index d553ba4..e8e48ef 100644
--- 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
index e69de29..a7a6d6c 100644
--- 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
index 0ac6715..d931eae 100644
--- 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
index d47c82a..c8d146e 100644
--- 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
index e7984cc..6d8df88 100644
--- 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
index b256465..320ce83 100644
--- 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
index 9eed86b..11c7a13 100644
--- 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
index 791d7a8..7ef3df1 100644
--- 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
index 902283c..28bd9fe 100644
--- 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