Fix tests

mozilla · Aug 29, 2019 · 28644a7 · 28644a7
1 parent 3c5aeb5
commit 28644a7
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/.travis/script b/.travis/script
@@ -11,5 +11,7 @@ fi
 
 if [[ "$TEST_SUITE" == "unittest" ]]; then
   # Run tests on all pushes
+  pushd tts_namespace
   python -m unittest
+  popd
 fi
diff --git a/tests/inputs/server_config.json b/tests/inputs/server_config.json
@@ -1,5 +1,5 @@
 {
-    "tts_path":"tests/outputs/",  // tts model root folder
+    "tts_path":"TTS/tests/outputs/",  // tts model root folder
     "tts_file":"checkpoint_10.pth.tar",     // tts checkpoint file
     "tts_config":"dummy_model_config.json",     // tts config.json file
     "tts_speakers": null,           // json file listing speaker ids. null if no speaker embedding.

diff --git a/utils/generic_utils.py b/utils/generic_utils.py
@@ -250,7 +250,7 @@ def set_init_dict(model_dict, checkpoint, c):
 
 def setup_model(num_chars, num_speakers, c):
     print(" > Using model: {}".format(c.model))
-    MyModel = importlib.import_module('models.' + c.model.lower())
+    MyModel = importlib.import_module('TTS.models.' + c.model.lower())
     MyModel = getattr(MyModel, c.model)
     if c.model.lower() in ["tacotron", "tacotrongst"]:
         model = MyModel(