fix server tests and pylint

pull/10/head
Eren Golge 2019-09-10 12:09:58 +02:00
parent 16d1f62afa
commit d1828c9573
2 changed files with 2 additions and 1 deletions

View File

@ -20,5 +20,6 @@ class DemoServerTest(unittest.TestCase):
def test_in_out(self):
self._create_random_model()
config = load_config(os.path.join(get_tests_input_path(), 'server_config.json'))
config['tts_path'] = get_tests_output_path()
synthesizer = Synthesizer(config)
synthesizer.tts("Better this test works!!")

View File

@ -32,7 +32,7 @@ def plot_spectrogram(linear_output, audio, fig_size=(16, 10)):
linear_output_ = linear_output.detach().cpu().numpy().squeeze()
else:
linear_output_ = linear_output
spectrogram = audio._denormalize(linear_output_)
spectrogram = audio._denormalize(linear_output_) # pylint: disable=protected-access
fig = plt.figure(figsize=fig_size)
plt.imshow(spectrogram.T, aspect="auto", origin="lower")
plt.colorbar()