diff --git a/matcha/models/baselightningmodule.py b/matcha/models/baselightningmodule.py index 3724888..f0de555 100644 --- a/matcha/models/baselightningmodule.py +++ b/matcha/models/baselightningmodule.py @@ -2,6 +2,7 @@ This is a base lightning module that can be used to train a model. The benefit of this abstraction is that all the logic outside of model definition can be reused for different models. """ + import inspect from abc import ABC from typing import Any, Dict diff --git a/matcha/text/__init__.py b/matcha/text/__init__.py index 71a4b57..452ea64 100644 --- a/matcha/text/__init__.py +++ b/matcha/text/__init__.py @@ -1,4 +1,5 @@ """ from https://github.com/keithito/tacotron """ + from matcha.text import cleaners from matcha.text.symbols import symbols diff --git a/matcha/text/symbols.py b/matcha/text/symbols.py index 7018df5..6249513 100644 --- a/matcha/text/symbols.py +++ b/matcha/text/symbols.py @@ -2,6 +2,7 @@ Defines the set of symbols used in text input to the model. """ + _pad = "_" _punctuation = ';:,.!?¡¿—…"«»“” ' _letters = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz" diff --git a/matcha/utils/generate_data_statistics.py b/matcha/utils/generate_data_statistics.py index 96a5382..fbfedf7 100644 --- a/matcha/utils/generate_data_statistics.py +++ b/matcha/utils/generate_data_statistics.py @@ -4,6 +4,7 @@ Parameters from hparam.py will be used """ + import argparse import json import os