import importlib


def load(strategy, tokenizer, cfg):
    try:
        load_fn = "load"
        if strategy.split(".")[-1].startswith("load_"):
            load_fn = strategy.split(".")[-1]
            strategy = ".".join(strategy.split(".")[:-1])
        m = importlib.import_module(f".{strategy}", "axolotl.prompt_strategies")
        fn = getattr(m, load_fn)
        return fn(tokenizer, cfg)
    except:
        pass