dump: Fix imports

revsic · Dec 19, 2022 · 277489b · 277489b
1 parent b1c1fea
commit 277489b
Showing 1 changed file with 8 additions and 8 deletions.
diff --git 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,