From d12b461d190596b0f23747ef926a462adf2ef9ad Mon Sep 17 00:00:00 2001
From: Wing Lian <wing@axolotl.ai>
Date: Sun, 10 Aug 2025 20:21:38 -0400
Subject: [PATCH] follow up fix for plugin registration (#3054) [skip ci]

---
 src/axolotl/cli/config.py        | 5 ++---
 src/axolotl/integrations/base.py | 4 ++--
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/src/axolotl/cli/config.py b/src/axolotl/cli/config.py
index da933929e..0f1245aed 100644
--- a/src/axolotl/cli/config.py
+++ b/src/axolotl/cli/config.py
@@ -153,15 +153,14 @@ def prepare_plugins(cfg: DictDefault):
         plugin_manager = PluginManager.get_instance()
         for plugin_name in cfg["plugins"]:
             plugin_manager.register(plugin_name)
+        for plugin in plugin_manager.plugins.values():
+            plugin.register(cfg)
 
 
 def plugin_set_cfg(cfg: DictDefault):
     if cfg.get("plugins"):
         plugin_manager = PluginManager.get_instance()
         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(
diff --git a/src/axolotl/integrations/base.py b/src/axolotl/integrations/base.py
index f43031287..94ee8d4b1 100644
--- a/src/axolotl/integrations/base.py
+++ b/src/axolotl/integrations/base.py
@@ -76,8 +76,8 @@ class BasePlugin:
     def __init__(self):
         """Initializes the BasePlugin."""
 
-    def register(self, cfg: DictDefault):  # pylint: disable=unused-argument
-        """Registers the plugin with the given configuration.
+    def register(self, cfg: dict):  # pylint: disable=unused-argument
+        """Registers the plugin with the given configuration as an unparsed dict.
 
         Args:
             cfg: The configuration for the plugin.