From 2b6b848f0a91a466bf218993d0b5bd2db5089366 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Sun, 25 Sep 2022 11:52:16 +0530 Subject: [PATCH] import fileprocessor --- enhancer/data/dataset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/enhancer/data/dataset.py b/enhancer/data/dataset.py index 797d691..f4e7e4a 100644 --- a/enhancer/data/dataset.py +++ b/enhancer/data/dataset.py @@ -8,9 +8,10 @@ from torch.utils.data import IterableDataset, DataLoader, Dataset import torch.nn.functional as F from typing import Optional +from enhancer.data.fileprocessor import Fileprocessor from enhancer.utils.random import create_unique_rng from enhancer.utils.io import Audio -from enhancer.utils import Fileprocessor, check_files +from enhancer.utils import check_files from enhancer.utils.config import Files class TrainDataset(IterableDataset):