From 277489bf815534c39182d7aa0483871f69a2a1f7 Mon Sep 17 00:00:00 2001
From: revsic <revsic99@gmail.com>
Date: Mon, 19 Dec 2022 13:36:15 +0000
Subject: [PATCH] dump: Fix imports

---
 utils/dump.py | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/utils/dump.py b/utils/dump.py
index 0f75ad0..3c95aca 100644
--- a/utils/dump.py
+++ b/utils/dump.py
@@ -6,10 +6,10 @@
 import numpy as np
 from tqdm import tqdm
 
-import speechset
+from .. import datasets
 
 
-class DumpReader(speechset.datasets.DataReader):
+class DumpReader(datasets.DataReader):
     """Dumped loader
     """
     def __init__(self, data_dir: str):
@@ -107,7 +107,7 @@ def dumper(args) -> Tuple[int, int, str, str]:
 
     @classmethod
     def dump(cls,
-             reader: speechset.datasets.DataReader,
+             reader: datasets.DataReader,
              out_dir: str,
              default_sid: int = -1,
              num_proc: Optional[int] = None,
@@ -173,11 +173,11 @@ def main():
         args = parser.parse_args()
 
         # hard code the reader
-        reader = speechset.datasets.ConcatReader([
-            speechset.datasets.LibriTTS('./datasets/LibriTTS/train-clean-100', args.sr),
-            speechset.datasets.LibriTTS('./datasets/LibriTTS/train-clean-360', args.sr),
-            speechset.datasets.LibriSpeech('./datasets/LibriSpeech/train-other-500', args.sr),
-            speechset.datasets.VCTK('./datasets/VCTK-Corpus', args.sr)])
+        reader = datasets.ConcatReader([
+            datasets.LibriTTS('./datasets/LibriTTS/train-clean-100', args.sr),
+            datasets.LibriTTS('./datasets/LibriTTS/train-clean-360', args.sr),
+            datasets.LibriSpeech('./datasets/LibriSpeech/train-other-500', args.sr),
+            datasets.VCTK('./datasets/VCTK-Corpus', args.sr)])
 
         DumpReader.dump(
             reader,