fix: import path for trainer builder and submodules

2025-05-15 15:49:37 +07:00
parent a1832953c4
commit 555190868a
12 changed files with 14 additions and 11 deletions
--- a/tests/core/test_trainer_builder.py
+++ b/tests/core/test_trainer_builder.py
@@ -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