tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

import from filepath if can't import_module

Browse Source
This commit is contained in:
Sunny Liu
2025-03-15 20:25:53 -04:00
committed by Sung Ching Liu
parent eadb16c709
commit ec52561a0c
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/axolotl/prompt_strategies/base.py
View File

@@ -36,7 +36,15 @@ def load(strategy, cfg, module_base=None, **kwargs):
module_base = ".".join(strategy.split(".")[:-2]) module_base = ".".join(strategy.split(".")[:-2])
strategy = strategy.split(".")[-2] strategy = strategy.split(".")[-2]
except ModuleNotFoundError: except ModuleNotFoundError:
strategy = "." + ".".join(strategy.split(".")[:-1]) try:
file_path = "/".join(strategy.split(".")[:-1]) + ".py"
module_name = strategy.split(".")[-2]
mod = import_from_path(module_name, file_path)
func = getattr(mod, load_fn)
return func(cfg, **kwargs)
except FileNotFoundError:
strategy = "." + ".".join(strategy.split(".")[:-1])
else: else:
strategy = "." + ".".join(strategy.split(".")[:-1]) strategy = "." + ".".join(strategy.split(".")[:-1])
mod = importlib.import_module(strategy, module_base) mod = importlib.import_module(strategy, module_base)