fix import error

Author: WeiNyn

src/data_reader/dataset.py

@@ -91,7 +91,7 @@ def _remove_entities(self, entities_list: Union[str, List[Dict[str, Any]]]) -> L
 
     sys.path.append(os.getcwd())
 
-    from src.DataReader.DataReader import make_dataframe
+    from src.data_reader.data_reader import make_dataframe
     from transformers import AutoTokenizer
 
     files = ["dataset/nlu_QnA_converted.yml", "dataset/nlu_QnA_converted.yml"]

src/models/classifier.py

@@ -176,8 +176,8 @@ def forward(
 
     sys.path.append(os.getcwd())
 
-    from src.DataReader.DataReader import make_dataframe
-    from src.DataReader.dataset import DIETClassifierDataset
+    from src.data_reader.data_reader import make_dataframe
+    from src.data_reader.dataset import DIETClassifierDataset
     from transformers import AutoTokenizer
 
     files = ["dataset/nlu_QnA_converted.yml", "dataset/nlu_QnA_converted.yml"]

src/models/trainer.py

@@ -58,8 +58,8 @@ def train(self):
 
     sys.path.append(os.getcwd())
 
-    from src.DataReader.DataReader import make_dataframe
-    from src.DataReader.dataset import DIETClassifierDataset
+    from src.data_reader.data_reader import make_dataframe
+    from src.data_reader.dataset import DIETClassifierDataset
     from src.models.classifier import DIETClassifier, DIETClassifierConfig
     from transformers import AutoTokenizer
 

src/models/wrapper.py

@@ -12,8 +12,8 @@
 
 from src.models.classifier import DIETClassifier, DIETClassifierConfig
 from src.models.trainer import DIETTrainer
-from src.DataReader.dataset import DIETClassifierDataset
-from src.DataReader.DataReader import make_dataframe
+from src.data_reader.dataset import DIETClassifierDataset
+from src.data_reader.data_reader import make_dataframe
 
 
 class DIETClassifierWrapper: