fix failure case in prompter loading

2025-02-04 08:43:33 -05:00
parent f9893e3842
commit 234cd8311e
1 changed files with 1 additions and 1 deletions
--- 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)