test_trainer_integration.py 222 Bytes
Newer Older
cmx's avatar
cmx committed
1
2
3
4
5
6
7
8
import pytest


def test_import():
    try:
        from liger_kernel.transformers.trainer_integration import _apply_liger_kernel  # noqa: F401
    except Exception:
        pytest.fail("Import _apply_liger_kernel fails")