Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor(punctuation): remove unused argument

This commit is contained in:
Enno Hermann 2023-11-29 22:08:49 +01:00
parent fee474cfe2
commit 87974f917a
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
TTS/tts/utils/text/punctuation.py
View File

@ -126,10 +126,10 @@ class Punctuation:
['This is', 'example'], ['.', '!'] -> "This is. example!" ['This is', 'example'], ['.', '!'] -> "This is. example!"
""" """
return cls._restore(text, puncs, 0) return cls._restore(text, puncs)
@classmethod @classmethod
def _restore(cls, text, puncs, num): # pylint: disable=too-many-return-statements def _restore(cls, text, puncs): # pylint: disable=too-many-return-statements
"""Auxiliary method for Punctuation.restore()""" """Auxiliary method for Punctuation.restore()"""
if not puncs: if not puncs:
return text return text
@ -141,18 +141,18 @@ class Punctuation:
current = puncs[0] current = puncs[0]
if current.position == PuncPosition.BEGIN: if current.position == PuncPosition.BEGIN:
return cls._restore([current.punc + text[0]] + text[1:], puncs[1:], num) return cls._restore([current.punc + text[0]] + text[1:], puncs[1:])
if current.position == PuncPosition.END: if current.position == PuncPosition.END:
return [text[0] + current.punc] + cls._restore(text[1:], puncs[1:], num + 1) return [text[0] + current.punc] + cls._restore(text[1:], puncs[1:])
# POSITION == MIDDLE # POSITION == MIDDLE
if len(text) == 1: # pragma: nocover if len(text) == 1: # pragma: nocover
# a corner case where the final part of an intermediate # a corner case where the final part of an intermediate
# mark (I) has not been phonemized # mark (I) has not been phonemized
return cls._restore([text[0] + current.punc], puncs[1:], num) return cls._restore([text[0] + current.punc], puncs[1:])
return cls._restore([text[0] + current.punc + text[1]] + text[2:], puncs[1:], num) return cls._restore([text[0] + current.punc + text[1]] + text[2:], puncs[1:])
# if __name__ == "__main__": # if __name__ == "__main__":