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

make plugin setup concise

Browse Source
This commit is contained in:
Wing Lian
2024-12-30 13:25:25 -05:00
parent cdfcd69afa
commit cba6165ae1
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/axolotl/integrations/base.py
View File

@@ -223,7 +223,17 @@ def load_plugin(plugin_name: str) -> BasePlugin:
module_name, class_name = plugin_name.rsplit(".", 1) module_name, class_name = plugin_name.rsplit(".", 1)
# import the module # import the module
module = importlib.import_module(module_name) try:
module = importlib.import_module(module_name)
except ModuleNotFoundError as orig_exc:
try:
if not module_name.startswith("axolotl.integrations."):
module = importlib.import_module("axolotl.integrations." + module_name)
else:
raise orig_exc
except ModuleNotFoundError as exc:
raise orig_exc from exc
# instantiate the class # instantiate the class
plugin_class = getattr(module, class_name) plugin_class = getattr(module, class_name)
# create an instance of the class # create an instance of the class