from pathlib import Path
import json


def load_config(train_or_apply: str) -> dict:
    """Load config"""
    config_file_path = Path(__file__).parent.resolve() / "config.json"
    with open(config_file_path, "r") as f:
        config = json.load(f)
    return config[train_or_apply]