diff --git a/Project_Model/Libs/BPE/Utils/json_utils.py b/Project_Model/Libs/BPE/Utils/json_utils.py index 716e93a..b98ac2f 100644 --- a/Project_Model/Libs/BPE/Utils/json_utils.py +++ b/Project_Model/Libs/BPE/Utils/json_utils.py @@ -2,15 +2,15 @@ import json from pathlib import Path -def save_json(vocabulary: dict, path: Path): +def save_json(dictionary: dict, path: Path): - json_string = json.dumps(vocabulary) + json_string = json.dumps(dictionary) FILE = open(path, "w") FILE.write(json_string) FILE.close() -def load_json(path: Path) -> dict[tuple[int, int], int]: +def load_json(path: Path) -> dict: FILE = open(path, "r") json_string = FILE.read() FILE.close()