c907ac173e
* adding pre-commit auto-update GH action and bumping plugin versions * running updated pre-commit plugins * sorry to revert, but pylint complained * Update .pre-commit-config.yaml Co-authored-by: Wing Lian <wing.lian@gmail.com> --------- Co-authored-by: Dan Saunders <dan@axolotl.ai> Co-authored-by: Wing Lian <wing.lian@gmail.com>
22 lines
642 B
Python
22 lines
642 B
Python
"""
|
|
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
|
|
"""
|
|
|
|
def test_import_causal_trainer(self):
|
|
from axolotl.core.trainer_builder 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
|
|
HFRLTrainerBuilder,
|
|
)
|