standardize wavenet imports

garrettgibo · Mar 17, 2021 · 7a3c3e8 · 7a3c3e8
1 parent f76893f
commit 7a3c3e8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/run.py b/run.py
@@ -3,15 +3,15 @@
 import argparse
 import json
 
-from wavenet import train
-from wavenet.utils import Mp3Converter, get_data, new_logger
+import wavenet
+import wavenet.utils as utils
 
-logger = new_logger("run")
+logger = utils.new_logger("run")
 
 TARGETS = {
-    "train": train,
-    "get-data": get_data,
-    "mp3-to-wav": Mp3Converter,
+    "train": wavenet.train_pipeline,
+    "get-data": utils.get_data,
+    "mp3-to-wav": utils.Mp3Converter,
 }
 
 CONFIGS = {

diff --git a/wavenet/__init__.py b/wavenet/__init__.py
@@ -4,7 +4,7 @@
 from wavenet.utils.data import WAVData, WAVDataLoader
 
 
-def train(
+def train_pipeline(
     mp3_to_wav_cfg: dict,
     dataset_cfg: dict,
     dataloader_cfg: dict,