Skip to content

Commit 79c22df

Browse files
committed
fix: miss import
1 parent 7f0ebf4 commit 79c22df

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/transformers/test_monkey_patch.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2592,6 +2592,7 @@ def test_apply_liger_kernel_to_instance_for_glm4():
25922592
@pytest.mark.skipif(not is_glm4_moe_available(), reason="glm4 module not available")
25932593
def test_apply_liger_kernel_to_instance_for_glm4_moe():
25942594
from transformers.models.glm4_moe.modeling_glm4_moe import Glm4MoeMLP
2595+
from transformers.models.glm4_moe.modeling_glm4_moe import Glm4MoeMOE
25952596

25962597
# Ensure any monkey patching is cleaned up for subsequent tests
25972598
with patch("transformers.models.glm4_moe.modeling_glm4_moe"):

0 commit comments

Comments
 (0)