fix: explicitly define exports

idiap · Nov 26, 2024 · c227252 · c227252
1 parent a63efbf
commit c227252
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/trainer/__init__.py b/trainer/__init__.py
@@ -1,7 +1,9 @@
 import importlib.metadata
 
 from trainer.config import TrainerArgs, TrainerConfig
-from trainer.model import *
-from trainer.trainer import *
+from trainer.model import TrainerModel
+from trainer.trainer import Trainer
 
 __version__ = importlib.metadata.version("coqui-tts-trainer")
+
+__all__ = ["TrainerArgs", "TrainerConfig", "Trainer", "TrainerModel"]