From a40e4848035f974ec45512fdf4f009609f6ceeb8 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sun, 25 May 2025 18:01:46 -0400 Subject: [PATCH] handle when no custom collator is used in plugins --- src/axolotl/integrations/base.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/axolotl/integrations/base.py b/src/axolotl/integrations/base.py index 7c2c773b1..f89dc5049 100644 --- a/src/axolotl/integrations/base.py +++ b/src/axolotl/integrations/base.py @@ -531,10 +531,11 @@ class PluginManager: object: The collator class, or None if none was found. """ for plugin in self.plugins.values(): - collator_cls, collator_kwargs = plugin.get_collator_cls_and_kwargs( + collator = plugin.get_collator_cls_and_kwargs( cfg, is_eval=is_eval ) - if collator_cls is not None: + if collator is not None: + collator_cls, collator_kwargs = collator return collator_cls, collator_kwargs return None