fix init import bug

.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,10 +19,10 @@ from pathlib import Path
 
 import tqdm
 import numpy as np
-from parakeet.audio.audio import AudioProcessor
-from parakeet.audio.spec_normalizer import LogMagnitude
-from parakeet.datasets.ljspeech import LJSpeechMetaData
-from parakeet.frontend.phonectic import English
+from parakeet.audio import AudioProcessor
+from parakeet.audio import LogMagnitude
+from parakeet.datasets import LJSpeechMetaData
+from parakeet.frontend import English
 
 from config import get_cfg_defaults
 

examples/transformer_tts/train.py

@@ -21,7 +21,7 @@ from paddle import distributed as dist
 from paddle.io import DataLoader, DistributedBatchSampler
 
 from parakeet.data import dataset
-from parakeet.frontend.phonectic import English
+from parakeet.frontend import English
 from parakeet.models.transformer_tts import TransformerTTS, TransformerTTSLoss
 from parakeet.utils import scheduler, mp_tools, display
 from parakeet.training.cli import default_argument_parser

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/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/models/transformer_tts.py

@@ -370,7 +370,7 @@ class CNNPostNet(nn.Layer):
 
 class TransformerTTS(nn.Layer):
     def __init__(self,
-                 frontend: parakeet.frontend.phonectic.Phonetics,
+                 frontend: parakeet.frontend.Phonetics,
                  d_encoder: int,
                  d_decoder: int,
                  d_mel: int,

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