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

fix: import path for trainer builder and submodules

Browse Source
This commit is contained in:
NanoCode012
2025-05-15 15:49:37 +07:00
parent a1832953c4
commit 555190868a
12 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/core/test_trainer_builder.py
View File

@@ -7,7 +7,8 @@ from pathlib import Path
import pytest
from axolotl.core.trainer_builder import HFCausalTrainerBuilder, HFRLTrainerBuilder
from axolotl.core.trainer_builder.rl import HFRLTrainerBuilder
from axolotl.core.trainer_builder.sft import HFCausalTrainerBuilder
from axolotl.utils.config import normalize_config
from axolotl.utils.dict import DictDefault
from axolotl.utils.models import load_model, load_tokenizer

4
tests/e2e/test_imports.py
View File

@@ -11,11 +11,11 @@ class TestImports(unittest.TestCase):
"""
def test_import_causal_trainer(self):
from axolotl.core.trainer_builder import ( # pylint: disable=unused-import # noqa: F401
from axolotl.core.trainer_builder.sft 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.trainer_builder.rl import ( # pylint: disable=unused-import # noqa: F401
HFRLTrainerBuilder,
)