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

refactor build() into smaller fns

Browse Source
This commit is contained in:
Dan Saunders
2025-05-12 20:36:52 +00:00
parent c7b6790614
commit 103edc7211
10 changed files with 1352 additions and 1256 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/core/test_trainer_builder.py
View File

@@ -1,10 +1,8 @@
"""
unit tests for axolotl.core.trainer_builder
"""
"""Unit tests for axolotl.core.trainers.builders"""
import pytest
from axolotl.core.trainer_builder import HFRLTrainerBuilder
from axolotl.core.trainers.builders import HFRLTrainerBuilder
from axolotl.utils.config import normalize_config
from axolotl.utils.dict import DictDefault
from axolotl.utils.models import load_model, load_tokenizer
@@ -53,9 +51,7 @@ def fixture_model(cfg, tokenizer):
class TestHFRLTrainerBuilder:
"""
TestCase class for DPO trainer builder
"""
"""Test case class for RL trainer builder"""
def test_build_training_arguments(self, cfg, model, tokenizer):
builder = HFRLTrainerBuilder(cfg, model, tokenizer)

12
tests/e2e/test_imports.py
View File

@@ -1,21 +1,21 @@
"""
test module to import various submodules that have historically broken due to dependency issues
"""Test module to import various submodules that have historically broken due to
dependency issues.
"""
import unittest
class TestImports(unittest.TestCase):
"""
Test class to import various submodules that have historically broken due to dependency issues
"""Test class to import various submodules that have historically broken due to
dependency issues.
"""
def test_import_causal_trainer(self):
from axolotl.core.trainer_builder import ( # pylint: disable=unused-import # noqa: F401
from axolotl.core.trainers.builders import ( # pylint: disable=unused-import # noqa: F401
HFCausalTrainerBuilder,
)
def test_import_rl_trainer(self):
from axolotl.core.trainer_builder import ( # pylint: disable=unused-import # noqa: F401
from axolotl.core.trainers.builders import ( # pylint: disable=unused-import # noqa: F401
HFRLTrainerBuilder,
)