From 234cd8311e55ef310c98c476019a7a0095efdfa7 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Tue, 4 Feb 2025 08:43:33 -0500 Subject: [PATCH] fix failure case in prompter loading --- src/axolotl/prompt_strategies/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/prompt_strategies/base.py b/src/axolotl/prompt_strategies/base.py index b3f2413ba..c146133fb 100644 --- a/src/axolotl/prompt_strategies/base.py +++ b/src/axolotl/prompt_strategies/base.py @@ -22,7 +22,7 @@ def load(strategy, cfg, module_base=None, **kwargs): module_base = ".".join(strategy.split(".")[:-2]) strategy = strategy.split(".")[-2] except ModuleNotFoundError: - pass + strategy = "." + ".".join(strategy.split(".")[:-1]) else: strategy = "." + ".".join(strategy.split(".")[:-1]) mod = importlib.import_module(strategy, module_base)