From dd407d882d3a60dbfd47f69e8b6d0ec1c98d5f2e Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Wed, 13 Mar 2024 18:45:25 +0900
Subject: [PATCH] Fix typing of arguments (Languages class)

---
 bert_gen.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/bert_gen.py b/bert_gen.py
index 1e4a92c..5619ee6 100644
--- a/bert_gen.py
+++ b/bert_gen.py
@@ -50,7 +50,9 @@ def process_line(x: tuple[str, bool]):
     tone = [int(i) for i in tone.split(" ")]
     word2ph = [int(i) for i in word2ph.split(" ")]
     word2ph = [i for i in word2ph]
-    phone, tone, language = cleaned_text_to_sequence(phone, tone, language_str)
+    phone, tone, language = cleaned_text_to_sequence(
+        phone, tone, Languages[language_str]
+    )
 
     if add_blank:
         phone = commons.intersperse(phone, 0)