Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Nov 5, 2024
1 parent d52fea5 commit 65d3c12
Show file tree
Hide file tree
Showing 8 changed files with 22 additions and 10 deletions.
4 changes: 3 additions & 1 deletion tests/nemo_text_processing/hi/test_cardinal.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,12 @@
import pytest
from parameterized import parameterized

from nemo_text_processing.text_normalization.normalize import Normalizer
from nemo_text_processing.inverse_text_normalization.inverse_normalize import InverseNormalizer
from nemo_text_processing.text_normalization.normalize import Normalizer

from ..utils import CACHE_DIR, parse_test_case_file


class TestCardinal:
normalizer = Normalizer(input_case='cased', lang='hi', cache_dir=CACHE_DIR, overwrite_cache=False)
inverse_normalizer = InverseNormalizer(lang='hi', cache_dir=CACHE_DIR, overwrite_cache=False)
Expand Down
4 changes: 3 additions & 1 deletion tests/nemo_text_processing/hi/test_date.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,12 @@
import pytest
from parameterized import parameterized

from nemo_text_processing.text_normalization.normalize import Normalizer
from nemo_text_processing.inverse_text_normalization.inverse_normalize import InverseNormalizer
from nemo_text_processing.text_normalization.normalize import Normalizer

from ..utils import CACHE_DIR, parse_test_case_file


class TestDate:
normalizer = Normalizer(input_case='cased', lang='hi', cache_dir=CACHE_DIR, overwrite_cache=False)
inverse_normalizer = InverseNormalizer(lang='hi', cache_dir=CACHE_DIR, overwrite_cache=False)
Expand Down
4 changes: 3 additions & 1 deletion tests/nemo_text_processing/hi/test_decimal.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,12 @@
import pytest
from parameterized import parameterized

from nemo_text_processing.text_normalization.normalize import Normalizer
from nemo_text_processing.inverse_text_normalization.inverse_normalize import InverseNormalizer
from nemo_text_processing.text_normalization.normalize import Normalizer

from ..utils import CACHE_DIR, parse_test_case_file


class TestDecimal:
normalizer = Normalizer(input_case='cased', lang='hi', cache_dir=CACHE_DIR, overwrite_cache=False)
inverse_normalizer = InverseNormalizer(lang='hi', cache_dir=CACHE_DIR, overwrite_cache=False)
Expand Down
4 changes: 3 additions & 1 deletion tests/nemo_text_processing/hi/test_fraction.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,12 @@
import pytest
from parameterized import parameterized

from nemo_text_processing.text_normalization.normalize import Normalizer
from nemo_text_processing.inverse_text_normalization.inverse_normalize import InverseNormalizer
from nemo_text_processing.text_normalization.normalize import Normalizer

from ..utils import CACHE_DIR, parse_test_case_file


class TestFraction:
normalizer = Normalizer(input_case='cased', lang='hi', cache_dir=CACHE_DIR, overwrite_cache=False)
inverse_normalizer = InverseNormalizer(lang='hi', cache_dir=CACHE_DIR, overwrite_cache=False)
Expand Down
4 changes: 3 additions & 1 deletion tests/nemo_text_processing/hi/test_measure.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,12 @@
import pytest
from parameterized import parameterized

from nemo_text_processing.text_normalization.normalize import Normalizer
from nemo_text_processing.inverse_text_normalization.inverse_normalize import InverseNormalizer
from nemo_text_processing.text_normalization.normalize import Normalizer

from ..utils import CACHE_DIR, parse_test_case_file


class TestMeasure:
normalizer = Normalizer(input_case='cased', lang='hi', cache_dir=CACHE_DIR, overwrite_cache=False)
inverse_normalizer = InverseNormalizer(lang='hi', cache_dir=CACHE_DIR, overwrite_cache=False)
Expand Down
4 changes: 3 additions & 1 deletion tests/nemo_text_processing/hi/test_money.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,12 @@
import pytest
from parameterized import parameterized

from nemo_text_processing.text_normalization.normalize import Normalizer
from nemo_text_processing.inverse_text_normalization.inverse_normalize import InverseNormalizer
from nemo_text_processing.text_normalization.normalize import Normalizer

from ..utils import CACHE_DIR, parse_test_case_file


class TestMoney:
normalizer = Normalizer(input_case='cased', lang='hi', cache_dir=CACHE_DIR, overwrite_cache=False)
inverse_normalizer = InverseNormalizer(lang='hi', cache_dir=CACHE_DIR, overwrite_cache=False)
Expand Down
4 changes: 3 additions & 1 deletion tests/nemo_text_processing/hi/test_time.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,10 +15,12 @@
import pytest
from parameterized import parameterized

from nemo_text_processing.text_normalization.normalize import Normalizer
from nemo_text_processing.inverse_text_normalization.inverse_normalize import InverseNormalizer
from nemo_text_processing.text_normalization.normalize import Normalizer

from ..utils import CACHE_DIR, parse_test_case_file


class TestTime:
normalizer = Normalizer(input_case='cased', lang='hi', cache_dir=CACHE_DIR, overwrite_cache=False)
inverse_normalizer = InverseNormalizer(lang='hi', cache_dir=CACHE_DIR, overwrite_cache=False)
Expand Down
4 changes: 1 addition & 3 deletions tools/text_processing_deployment/pynini_export.py
Original file line number Diff line number Diff line change
Expand Up @@ -213,9 +213,7 @@ def parse_args():
from nemo_text_processing.text_normalization.hi.taggers.tokenize_and_classify import (
ClassifyFst as TNClassifyFst,
)
from nemo_text_processing.text_normalization.hi.verbalizers.verbalize import (
VerbalizeFst as TNVerbalizeFst,
)
from nemo_text_processing.text_normalization.hi.verbalizers.verbalize import VerbalizeFst as TNVerbalizeFst
elif args.language == 'hu':
from nemo_text_processing.text_normalization.hu.taggers.tokenize_and_classify import (
ClassifyFst as TNClassifyFst,
Expand Down

0 comments on commit 65d3c12

Please sign in to comment.