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

handle when no custom collator is used in plugins

Browse Source
This commit is contained in:
Wing Lian
2025-05-25 18:01:46 -04:00
parent 9899c924f9
commit a40e484803
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -531,10 +531,11 @@ class PluginManager:
object: The collator class, or None if none was found. object: The collator class, or None if none was found.
""" """
for plugin in self.plugins.values(): 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 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 collator_cls, collator_kwargs
return None return None