From 9b74298328aa2a544aa5b7f7616cab06e3a7c6a1 Mon Sep 17 00:00:00 2001 From: Sung Ching Liu <22844540+bursteratom@users.noreply.github.com> Date: Wed, 2 Apr 2025 14:33:52 -0400 Subject: [PATCH] Update src/axolotl/prompt_strategies/base.py Co-authored-by: Wing Lian --- src/axolotl/prompt_strategies/base.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/axolotl/prompt_strategies/base.py b/src/axolotl/prompt_strategies/base.py index 3258effa7..12b436a0c 100644 --- a/src/axolotl/prompt_strategies/base.py +++ b/src/axolotl/prompt_strategies/base.py @@ -41,7 +41,10 @@ def load(strategy, cfg, module_base=None, **kwargs): module_name = strategy.split(".")[-2] mod = import_from_path(module_name, file_path) func = getattr(mod, load_fn) - return func(cfg, **kwargs) + if func is None: + strategy = "." + ".".join(strategy.split(".")[:-1]) + else: + return func(cfg, **kwargs) except FileNotFoundError: strategy = "." + ".".join(strategy.split(".")[:-1])