From 04df0a3d9f4a38be0c4c5d617365003d115c9ed1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Tue, 30 Nov 2021 15:50:18 +0100
Subject: [PATCH] =?UTF-8?q?Refactor=20TTSDataset=20=E2=9A=A1=EF=B8=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 TTS/tts/datasets/dataset.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/TTS/tts/datasets/dataset.py b/TTS/tts/datasets/dataset.py
index 9b78ddba..9de40c2b 100644
--- a/TTS/tts/datasets/dataset.py
+++ b/TTS/tts/datasets/dataset.py
@@ -216,8 +216,8 @@ class TTSDataset(Dataset):
         elif len(item) == 3:
             text, wav_file, speaker_name = item
         else:
-            raise ValueError(" [!] Dataset cannot parse the sample.")
-        return text, wav_file, speaker_name, language_name, attn_file
+            token_ids = self.tokenizer.text_to_ids(text)
+        return token_ids
 
     def load_data(self, idx):
         item = self.samples[idx]