Merge pull request #152 from yt605155624/fix_transformer_tts

fix import bug in transformer tts

.flake8

@@ -42,6 +42,10 @@ ignore =
     # these ignores are from flake8-comprehensions; please fix!
     C400,C401,C402,C403,C404,C405,C407,C411,C413,C414,C415
 
+
+per-file-ignores =
+    */__init__.py: F401
+
 # Specify the list of error codes you wish Flake8 to report.
 select =
     E,

examples/transformer_tts/preprocess.py

@@ -19,9 +19,9 @@ from pathlib import Path
 
 import tqdm
 import numpy as np
-
+from parakeet.audio import AudioProcessor
+from parakeet.audio import LogMagnitude
 from parakeet.datasets import LJSpeechMetaData
-from parakeet.audio import AudioProcessor, LogMagnitude
 from parakeet.frontend import English
 
 from config import get_cfg_defaults

parakeet/__init__.py

@@ -13,3 +13,11 @@
 # limitations under the License.
 
 __version__ = "0.0.0"
+import logging
+from . import data
+from . import datasets
+from . import frontend
+from . import models
+from . import modules
+from . import training
+from . import utils

parakeet/audio/__init__.py

@@ -11,3 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from .audio import AudioProcessor
+from .spec_normalizer import LogMagnitude
+from .spec_normalizer import NormalizerBase

parakeet/data/__init__.py

@@ -13,3 +13,6 @@
 # limitations under the License.
 """Parakeet's infrastructure for data processing.
 """
+
+from .dataset import *
+from .batch import *

parakeet/datasets/__init__.py

@@ -11,3 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from .common import *
+from .ljspeech import *

parakeet/frontend/__init__.py

@@ -11,3 +11,11 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from .cn_normalization import *
+from .generate_lexicon import *
+from .normalizer import *
+from .phonectic import *
+from .punctuation import *
+from .tone_sandhi import *
+from .vocab import *

parakeet/frontend/cn_normalization/__init__.py

@@ -11,3 +11,5 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from parakeet.frontend.cn_normalization.text_normlization import *

parakeet/frontend/normalizer/__init__.py

@@ -11,3 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from parakeet.frontend.normalizer.normalizer import *
+from parakeet.frontend.normalizer.numbers import *

parakeet/frontend/phonectic.py

@@ -17,7 +17,7 @@ from abc import abstractmethod
 from g2p_en import G2p
 from g2pM import G2pM
 
-from parakeet.frontend import Vocab
+from parakeet.frontend.vocab import Vocab
 from parakeet.frontend.normalizer.normalizer import normalize
 from parakeet.frontend.punctuation import get_punctuations
 

parakeet/models/__init__.py

@@ -11,3 +11,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from .fastspeech2 import *
+from .tacotron2 import *
+from .transformer_tts import *
+from .waveflow import *

parakeet/modules/__init__.py

@@ -11,3 +11,11 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from .attention import *
+from .conv import *
+from .geometry import *
+from .losses import *
+from .masking import *
+from .positional_encoding import *
+from .transformer import *

parakeet/training/__init__.py

@@ -11,3 +11,6 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from .cli import *
+from .experiment import *

parakeet/utils/__init__.py

@@ -11,3 +11,9 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from . import checkpoint
+from . import display
+from . import layer_tools
+from . import mp_tools
+from . import scheduler