| import os | |
| import json | |
| import logging | |
| from datetime import datetime | |
| def load_json(file_name: str): | |
| if isinstance(file_name, str) and file_name.endswith("json"): | |
| with open(file_name, 'r') as file: | |
| data = json.load(file) | |
| else: | |
| raise ValueError("The file path you passed in is not a json file path.") | |
| return data | |
| def init_logger(outputs_dir): | |
| current_time = datetime.now().strftime("%b%d_%H-%M-%S") | |
| os.makedirs(os.path.join(outputs_dir, "logs"), exist_ok=True) | |
| log_path = os.path.join(outputs_dir, "logs", "{}.txt".format(current_time)) | |
| logging.basicConfig( | |
| level=logging.INFO, | |
| format="%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s || %(message)s", | |
| handlers=[logging.StreamHandler(), logging.FileHandler(log_path)], | |
| ) | |