Apply black formatter

style_bert_vits2/nlp/japanese/g2p.py

@@ -10,9 +10,7 @@ from style_bert_vits2.nlp.symbols import PUNCTUATIONS
 
 
 def g2p(
-    norm_text: str,
-    use_jp_extra: bool = True,
-    raise_yomi_error: bool = False
+    norm_text: str, use_jp_extra: bool = True, raise_yomi_error: bool = False
 ) -> tuple[list[str], list[int], list[int]]:
     """
     他で使われるメインの関数。`normalize_text()` で正規化された `norm_text` を受け取り、
@@ -93,8 +91,7 @@ def g2p(
 
 
 def text_to_sep_kata(
-    norm_text: str,
-    raise_yomi_error: bool = False
+    norm_text: str, raise_yomi_error: bool = False
 ) -> tuple[list[str], list[str]]:
     """
     `normalize_text` で正規化済みの `norm_text` を受け取り、それを単語分割し、
@@ -212,7 +209,9 @@ def __g2phone_tone_wo_punct(text: str) -> list[tuple[str, int]]:
     return result
 
 
-def __pyopenjtalk_g2p_prosody(text: str, drop_unvoiced_vowels: bool = True) -> list[str]:
+def __pyopenjtalk_g2p_prosody(
+    text: str, drop_unvoiced_vowels: bool = True
+) -> list[str]:
     """
     ESPnet の実装から引用、変更点無し。「ん」は「N」なことに注意。
     ref: https://github.com/espnet/espnet/blob/master/espnet2/text/phoneme_tokenizer.py
@@ -414,8 +413,7 @@ def __kata_to_phoneme_list(text: str) -> list[str]:
 
 
 def __align_tones(
-    phones_with_punct: list[str],
-    phone_tone_list: list[tuple[str, int]]
+    phones_with_punct: list[str], phone_tone_list: list[tuple[str, int]]
 ) -> list[tuple[str, int]]:
     """
     例: …私は、、そう思う。

style_bert_vits2/nlp/japanese/g2p_utils.py

@@ -34,11 +34,13 @@ def phone_tone2kata_tone(phone_tone: list[tuple[str, int]]) -> list[tuple[str, i
     """
 
     # 子音の集合
-    CONSONANTS = set([
-        consonant
-        for consonant, _ in MORA_KATA_TO_MORA_PHONEMES.values()
-        if consonant is not None
-    ])
+    CONSONANTS = set(
+        [
+            consonant
+            for consonant, _ in MORA_KATA_TO_MORA_PHONEMES.values()
+            if consonant is not None
+        ]
+    )
 
     phone_tone = phone_tone[1:]  # 最初の("_", 0)を無視
     phones = [phone for phone, _ in phone_tone]

style_bert_vits2/nlp/japanese/pyopenjtalk_worker/__init__.py

@@ -25,6 +25,7 @@ def run_frontend(text: str) -> list[dict[str, Any]]:
     else:
         # without worker
         import pyopenjtalk
+
         return pyopenjtalk.run_frontend(text)
 
 
@@ -36,6 +37,7 @@ def make_label(njd_features: Any) -> list[str]:
     else:
         # without worker
         import pyopenjtalk
+
         return pyopenjtalk.make_label(njd_features)
 
 
@@ -45,6 +47,7 @@ def mecab_dict_index(path: str, out_path: str, dn_mecab: Optional[str] = None) -
     else:
         # without worker
         import pyopenjtalk
+
         pyopenjtalk.mecab_dict_index(path, out_path, dn_mecab)
 
 
@@ -54,6 +57,7 @@ def update_global_jtalk_with_user_dict(path: str) -> None:
     else:
         # without worker
         import pyopenjtalk
+
         pyopenjtalk.update_global_jtalk_with_user_dict(path)
 
 
@@ -63,6 +67,7 @@ def unset_user_dict() -> None:
     else:
         # without worker
         import pyopenjtalk
+
         pyopenjtalk.unset_user_dict()
 
 
@@ -102,7 +107,12 @@ def initialize_worker(port: int = WORKER_PORT) -> None:
         else:
             # align with Windows behavior
             # start_new_session is same as specifying setsid in preexec_fn
-            subprocess.Popen(args, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL, start_new_session=True)
+            subprocess.Popen(
+                args,
+                stdout=subprocess.DEVNULL,
+                stderr=subprocess.DEVNULL,
+                start_new_session=True,
+            )
 
         # wait until server listening
         count = 0

style_bert_vits2/nlp/japanese/pyopenjtalk_worker/worker_client.py

@@ -2,12 +2,15 @@ import socket
 from typing import Any, cast
 
 from style_bert_vits2.logging import logger
-from style_bert_vits2.nlp.japanese.pyopenjtalk_worker.worker_common import RequestType, receive_data, send_data
+from style_bert_vits2.nlp.japanese.pyopenjtalk_worker.worker_common import (
+    RequestType,
+    receive_data,
+    send_data,
+)
 
 
 class WorkerClient:
-    """ pyopenjtalk worker client """
-
+    """pyopenjtalk worker client"""
 
     def __init__(self, port: int) -> None:
         sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
@@ -16,19 +19,15 @@ class WorkerClient:
         sock.connect((socket.gethostname(), port))
         self.sock = sock
 
-
     def __enter__(self) -> "WorkerClient":
         return self
 
-
     def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None:
         self.close()
 
-
     def close(self) -> None:
         self.sock.close()
 
-
     def dispatch_pyopenjtalk(self, func: str, *args: Any, **kwargs: Any) -> Any:
         data = {
             "request-type": RequestType.PYOPENJTALK,
@@ -43,7 +42,6 @@ class WorkerClient:
         logger.trace(f"client received response: {response}")
         return response.get("return")
 
-
     def status(self) -> int:
         data = {"request-type": RequestType.STATUS}
         logger.trace(f"client sends request: {data}")
@@ -53,7 +51,6 @@ class WorkerClient:
         logger.trace(f"client received response: {response}")
         return cast(int, response.get("client-count"))
 
-
     def quit_server(self) -> None:
         data = {"request-type": RequestType.QUIT_SERVER}
         logger.trace(f"client sends request: {data}")

style_bert_vits2/nlp/japanese/pyopenjtalk_worker/worker_server.py

@@ -26,14 +26,12 @@ PYOPENJTALK_FUNC_DICT = {
 
 
 class WorkerServer:
-    """ pyopenjtalk worker server """
-
+    """pyopenjtalk worker server"""
 
     def __init__(self) -> None:
         self.client_count: int = 0
         self.quit: bool = False
 
-
     def handle_request(self, request: dict[str, Any]) -> dict[str, Any]:
         request_type = None
         try:
@@ -70,7 +68,6 @@ class WorkerServer:
 
         return response
 
-
     def start_server(self, port: int, no_client_timeout: int = 30) -> None:
         logger.info("start pyopenjtalk worker server")
         with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as server_socket:

style_bert_vits2/nlp/japanese/user_dict/__init__.py

@@ -18,7 +18,11 @@ from fastapi import HTTPException
 from style_bert_vits2.constants import DEFAULT_USER_DICT_DIR
 from style_bert_vits2.nlp.japanese import pyopenjtalk_worker as pyopenjtalk
 from style_bert_vits2.nlp.japanese.user_dict.word_model import UserDictWord, WordTypes
-from style_bert_vits2.nlp.japanese.user_dict.part_of_speech_data import MAX_PRIORITY, MIN_PRIORITY, part_of_speech_data
+from style_bert_vits2.nlp.japanese.user_dict.part_of_speech_data import (
+    MAX_PRIORITY,
+    MIN_PRIORITY,
+    part_of_speech_data,
+)
 
 # root_dir = engine_root()
 # save_dir = get_save_dir()
@@ -26,9 +30,13 @@ from style_bert_vits2.nlp.japanese.user_dict.part_of_speech_data import MAX_PRIO
 # if not save_dir.is_dir():
 #     save_dir.mkdir(parents=True)
 
-default_dict_path = DEFAULT_USER_DICT_DIR / "default.csv"  # VOICEVOXデフォルト辞書ファイルのパス
+default_dict_path = (
+    DEFAULT_USER_DICT_DIR / "default.csv"
+)  # VOICEVOXデフォルト辞書ファイルのパス
 user_dict_path = DEFAULT_USER_DICT_DIR / "user_dict.json"  # ユーザー辞書ファイルのパス
-compiled_dict_path = DEFAULT_USER_DICT_DIR / "user.dic"  # コンパイル済み辞書ファイルのパス
+compiled_dict_path = (
+    DEFAULT_USER_DICT_DIR / "user.dic"
+)  # コンパイル済み辞書ファイルのパス
 
 
 # # 同時書き込みの制御