From eadb16c70928335fd81b5b09acbac829247de7d7 Mon Sep 17 00:00:00 2001 From: Sunny Liu Date: Wed, 12 Mar 2025 00:19:09 -0400 Subject: [PATCH] test import-wihtin-import relative path --- src/axolotl/prompt_strategies/base.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/src/axolotl/prompt_strategies/base.py b/src/axolotl/prompt_strategies/base.py index c146133fb..afd8d01fa 100644 --- a/src/axolotl/prompt_strategies/base.py +++ b/src/axolotl/prompt_strategies/base.py @@ -4,10 +4,24 @@ module for base dataset transform strategies import importlib import logging +import sys LOG = logging.getLogger("axolotl") +def import_from_path(module_name, file_path): + spec = importlib.util.spec_from_file_location(module_name, file_path) + if spec is None: + raise ImportError(f"Could not create module spec for: {file_path}") + module = importlib.util.module_from_spec(spec) + + sys.modules[module_name] = module + loader = importlib.machinery.SourceFileLoader(module_name, file_path) + spec.loader = loader + loader.exec_module(module) + return module + + def load(strategy, cfg, module_base=None, **kwargs): try: if len(strategy.split(".")) == 1: