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

fix: address renamed cfg

Browse Source
This commit is contained in:
NanoCode012
2025-05-23 13:34:04 +07:00
parent 29dbf98ed2
commit af22816363
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/core/test_trainer_builder.py
View File

@@ -347,20 +347,19 @@ class TestTrainerClsPlugin:
TestCase class for trainer builder with plugin TestCase class for trainer builder with plugin
""" """
def test_trainer_cls_is_not_none_with_plugin(self, cfg, model, tokenizer): def test_trainer_cls_is_not_none_with_plugin(self, kto_cfg, model, tokenizer):
""" """
Test that the trainer cls is not none with plugin Test that the trainer cls is not none with plugin
Fixes #2693 Fixes #2693
""" """
cfg.plugins = ["axolotl.integrations.liger.LigerPlugin"] kto_cfg.plugins = ["axolotl.integrations.liger.LigerPlugin"]
cfg.rl = RLType.KTO
# Expected AttributeError as we don't pass regular model configs to RL trainer builder # Expected AttributeError as we don't pass regular model configs to RL trainer builder
# If it throws `TypeError: None is not a callable object`, trainer_cls could be None # If it throws `TypeError: None is not a callable object`, trainer_cls could be None
with pytest.raises( with pytest.raises(
AttributeError, match=r".*'tuple' object has no attribute 'config'.*" AttributeError, match=r".*'tuple' object has no attribute 'config'.*"
): ):
builder = HFRLTrainerBuilder(cfg, model, tokenizer) builder = HFRLTrainerBuilder(kto_cfg, model, tokenizer)
builder.build(100) builder.build(100)