diff --git a/enhancer/models/model.py b/enhancer/models/model.py index 5d1cd3e..999dc6c 100644 --- a/enhancer/models/model.py +++ b/enhancer/models/model.py @@ -15,7 +15,7 @@ from pathlib import Path from enhancer import __version__ from enhancer.data.dataset import EnhancerDataset from enhancer.utils.io import Audio -from enhancer.utils.loss import Avergeloss +from enhancer.loss import Avergeloss from enhancer.inference import Inference CACHE_DIR = "" diff --git a/enhancer/utils/__init__.py b/enhancer/utils/__init__.py index 39ae32a..a8ae539 100644 --- a/enhancer/utils/__init__.py +++ b/enhancer/utils/__init__.py @@ -1,3 +1,2 @@ -from enhancer.utils.fileprocessor import Fileprocessor from enhancer.utils.utils import check_files from enhancer.utils.io import Audio \ No newline at end of file diff --git a/tests/utils_test.py b/tests/utils_test.py index 81de1a4..413bfac 100644 --- a/tests/utils_test.py +++ b/tests/utils_test.py @@ -5,7 +5,7 @@ import numpy as np from enhancer.utils.io import Audio from enhancer.utils.config import Files -from enhancer.utils.fileprocessor import Fileprocessor +from enhancer.data.fileprocessor import Fileprocessor def test_io_channel():