From d0f8ad2f37bacaac294729bfa619a24a07456d8e Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Wed, 5 Oct 2022 20:36:50 +0530
Subject: [PATCH] dataset

---
 enhancer/data/dataset.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/enhancer/data/dataset.py b/enhancer/data/dataset.py
index d194167..95c73a1 100644
--- a/enhancer/data/dataset.py
+++ b/enhancer/data/dataset.py
@@ -1,16 +1,17 @@
-import multiprocessing
 import math
+import multiprocessing
 import os
-import pytorch_lightning as pl
-from torch.utils.data import IterableDataset, DataLoader, Dataset
-import torch.nn.functional as F
 from typing import Optional
 
+import pytorch_lightning as pl
+import torch.nn.functional as F
+from torch.utils.data import DataLoader, Dataset, IterableDataset
+
 from enhancer.data.fileprocessor import Fileprocessor
-from enhancer.utils.random import create_unique_rng
-from enhancer.utils.io import Audio
 from enhancer.utils import check_files
 from enhancer.utils.config import Files
+from enhancer.utils.io import Audio
+from enhancer.utils.random import create_unique_rng
 
 
 class TrainDataset(IterableDataset):