From 72ac2bfa09fbfd1775ba017aa13875e9e7bcaa55 Mon Sep 17 00:00:00 2001
From: Aarni Koskela <akx@iki.fi>
Date: Tue, 26 Sep 2023 23:08:45 +0300
Subject: [PATCH] Get rid of some star imports

---
 TTS/vc/modules/freevc/speaker_encoder/audio.py         |  8 +++++++-
 .../modules/freevc/speaker_encoder/speaker_encoder.py  | 10 +++++++++-
 recipes/bel-alex73/train_hifigan.py                    |  2 +-
 tests/data_tests/test_loader.py                        |  3 ++-
 4 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/TTS/vc/modules/freevc/speaker_encoder/audio.py b/TTS/vc/modules/freevc/speaker_encoder/audio.py
index 52f6fd08..30a77adc 100644
--- a/TTS/vc/modules/freevc/speaker_encoder/audio.py
+++ b/TTS/vc/modules/freevc/speaker_encoder/audio.py
@@ -7,7 +7,13 @@ import librosa
 import numpy as np
 from scipy.ndimage.morphology import binary_dilation
 
-from TTS.vc.modules.freevc.speaker_encoder.hparams import *
+from TTS.vc.modules.freevc.speaker_encoder.hparams import (
+    audio_norm_target_dBFS,
+    mel_n_channels,
+    mel_window_length,
+    mel_window_step,
+    sampling_rate,
+)
 
 int16_max = (2**15) - 1
 
diff --git a/TTS/vc/modules/freevc/speaker_encoder/speaker_encoder.py b/TTS/vc/modules/freevc/speaker_encoder/speaker_encoder.py
index 2e21a14f..d18c098f 100644
--- a/TTS/vc/modules/freevc/speaker_encoder/speaker_encoder.py
+++ b/TTS/vc/modules/freevc/speaker_encoder/speaker_encoder.py
@@ -8,7 +8,15 @@ from torch import nn
 
 from TTS.utils.io import load_fsspec
 from TTS.vc.modules.freevc.speaker_encoder import audio
-from TTS.vc.modules.freevc.speaker_encoder.hparams import *
+from TTS.vc.modules.freevc.speaker_encoder.hparams import (
+    mel_n_channels,
+    mel_window_step,
+    model_embedding_size,
+    model_hidden_size,
+    model_num_layers,
+    partials_n_frames,
+    sampling_rate,
+)
 
 
 class SpeakerEncoder(nn.Module):
diff --git a/recipes/bel-alex73/train_hifigan.py b/recipes/bel-alex73/train_hifigan.py
index 3e740b2f..1ddf48c2 100644
--- a/recipes/bel-alex73/train_hifigan.py
+++ b/recipes/bel-alex73/train_hifigan.py
@@ -5,7 +5,7 @@ from trainer import Trainer, TrainerArgs
 
 from TTS.tts.configs.shared_configs import BaseAudioConfig
 from TTS.utils.audio import AudioProcessor
-from TTS.vocoder.configs.hifigan_config import *
+from TTS.vocoder.configs.hifigan_config import HifiganConfig
 from TTS.vocoder.datasets.preprocess import load_wav_data
 from TTS.vocoder.models.gan import GAN
 
diff --git a/tests/data_tests/test_loader.py b/tests/data_tests/test_loader.py
index cbd98fc0..adb272e1 100644
--- a/tests/data_tests/test_loader.py
+++ b/tests/data_tests/test_loader.py
@@ -6,9 +6,10 @@ import numpy as np
 import torch
 from torch.utils.data import DataLoader
 
+from TTS.tts.datasets.dataset import TTSDataset
 from tests import get_tests_data_path, get_tests_output_path
 from TTS.tts.configs.shared_configs import BaseDatasetConfig, BaseTTSConfig
-from TTS.tts.datasets import TTSDataset, load_tts_samples
+from TTS.tts.datasets import load_tts_samples
 from TTS.tts.utils.text.tokenizer import TTSTokenizer
 from TTS.utils.audio import AudioProcessor