Bump deepspeed 20240727 (#1790)

* pin deepspeed to 0.14.4 otherwise it doesn't play nice with trl

* Add test to import to try to trigger import dependencies
This commit is contained in:
Wing Lian
2024-07-27 10:24:11 -04:00
committed by GitHub
parent 6a9cfec222
commit 22680913f3
3 changed files with 22 additions and 2 deletions

20
tests/e2e/test_imports.py Normal file
View File

@@ -0,0 +1,20 @@
"""
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,
)