Autism_QA / configs.py
A7m0d's picture
Upload folder using huggingface_hub
712579e verified
import yaml
import os
from typing import List
from logger.custom_logger import CustomLoggerTracker
# ------------------------------
# Custom Logger Initialization
# ------------------------------
custom_log = CustomLoggerTracker()
logger = custom_log.get_logger("configs")
logger.info("Logger initialized for Configs Query module")
def load_yaml_config(file_dir: str):
with open(file_dir, "r") as f:
config = yaml.safe_load(f)
if config is None:
return {}
return config
def _log(process_log: List[str], message: str, level: str = "info") -> None:
"""Append to process_log AND send to the central logger."""
process_log.append(message)
if level == "error":
logger.error(message)
elif level == "warning":
logger.warning(message)
else:
logger.info(message)
if __name__=="__main__":
configs = load_yaml_config("config.yaml")
logger.info(f"configs: {configs}")