Update data_loader.py

chinese-poetry · May 10, 2023 · 717029a · 717029a
1 parent aa65d2f
commit 717029a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/loader/data_loader.py b/loader/data_loader.py
@@ -5,7 +5,7 @@
 DATAS_CONFIG = "./loader/datas.json"
 
 
-class INGDataLoader():
+class PlainDataLoader():
     def __init__(self, config_path: str=DATAS_CONFIG) -> None:
         self._path = config_path
         with open(config_path, 'r', encoding='utf-8') as config:
@@ -58,7 +58,7 @@ def extract_with_ids(self, ids: list) -> list:
 
 
 if __name__ == "__main__":
-    loader = INGDataLoader()
+    loader = PlainDataLoader()
     print(loader.id_table)
     print(
         loader.body_extractor("wudai-huajianji")[-1]