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

follow up fix for plugin registration (#3054) [skip ci]

Browse Source
This commit is contained in:
Wing Lian
2025-08-10 20:21:38 -04:00
committed by GitHub
parent d6b81b3683
commit d12b461d19
2 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/axolotl/cli/config.py
View File

@@ -153,15 +153,14 @@ def prepare_plugins(cfg: DictDefault):
plugin_manager = PluginManager.get_instance() plugin_manager = PluginManager.get_instance()
for plugin_name in cfg["plugins"]: for plugin_name in cfg["plugins"]:
plugin_manager.register(plugin_name) plugin_manager.register(plugin_name)
for plugin in plugin_manager.plugins.values():
plugin.register(cfg)
def plugin_set_cfg(cfg: DictDefault): def plugin_set_cfg(cfg: DictDefault):
if cfg.get("plugins"): if cfg.get("plugins"):
plugin_manager = PluginManager.get_instance() plugin_manager = PluginManager.get_instance()
plugin_manager.cfg = cfg plugin_manager.cfg = cfg
# now that we have the finalized cfg, register the plugins individually
for plugin in plugin_manager.plugins.values():
plugin.register(cfg)
def load_cfg( def load_cfg(

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

@@ -76,8 +76,8 @@ class BasePlugin:
def __init__(self): def __init__(self):
"""Initializes the BasePlugin.""" """Initializes the BasePlugin."""
def register(self, cfg: DictDefault): # pylint: disable=unused-argument def register(self, cfg: dict): # pylint: disable=unused-argument
"""Registers the plugin with the given configuration. """Registers the plugin with the given configuration as an unparsed dict.
Args: Args:
cfg: The configuration for the plugin. cfg: The configuration for the plugin.