tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 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

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,
)