From 38e38dc7dfe440cf051600d490a75f28e3975358 Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Wed, 28 Sep 2022 11:51:58 +0530
Subject: [PATCH] check pythonpath

---
 cli/train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cli/train.py b/cli/train.py
index bc5e287..9ed1cd0 100644
--- a/cli/train.py
+++ b/cli/train.py
@@ -4,6 +4,7 @@ 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):