administrator/chinese-poetry
1
0
Fork 0
mirror of https://kkgithub.com/chinese-poetry/chinese-poetry.git synced 2026-04-11 19:17:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update data_loader.py

Browse Source
This commit is contained in:
SanBingYouYong-S.Zhang
2023-05-10 17:34:45 +01:00
committed by GitHub
parent aa65d2fa5e
commit 717029ade7
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
loader/data_loader.py
View File

@@ -5,7 +5,7 @@ import os
DATAS_CONFIG = "./loader/datas.json" DATAS_CONFIG = "./loader/datas.json"
class INGDataLoader(): class PlainDataLoader():
def __init__(self, config_path: str=DATAS_CONFIG) -> None: def __init__(self, config_path: str=DATAS_CONFIG) -> None:
self._path = config_path self._path = config_path
with open(config_path, 'r', encoding='utf-8') as config: with open(config_path, 'r', encoding='utf-8') as config:
@@ -58,7 +58,7 @@ class INGDataLoader():
if __name__ == "__main__": if __name__ == "__main__":
loader = INGDataLoader() loader = PlainDataLoader()
print(loader.id_table) print(loader.id_table)
print( print(
loader.body_extractor("wudai-huajianji")[-1] loader.body_extractor("wudai-huajianji")[-1]