From f4e625e41289396d8e1008c66e25bee0fbacef37 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Wed, 28 Sep 2022 12:49:59 +0530
Subject: [PATCH] rmv import

---
 cli/train.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/cli/train.py b/cli/train.py
index 9ed1cd0..9aa497d 100644
--- a/cli/train.py
+++ b/cli/train.py
@@ -4,8 +4,6 @@ from omegaconf import DictConfig
 from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping
 from pytorch_lightning.loggers import MLFlowLogger
 
-from enhancer.data.dataset import EnhancerDataset
-
 @hydra.main(config_path="train_config",config_name="config")
 def main(config: DictConfig):