Skip to content

Commit bcfb465

Browse files
committed
feat(tests): remove unused import in sequence parallel test
Remove `_get_sp_group_from_device_mesh` import from test file as it is no longer used in the test, cleaning up imports and improving code clarity.
1 parent c6ace7f commit bcfb465

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

tests/sequence_parallel/test_sequence_parallel_single_attention.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,6 @@
1919

2020
from twinkle.model.transformers.strategy.sequence_parallel import (
2121
DistributedAttention,
22-
_get_sp_group_from_device_mesh,
2322
sequence_parallel,
2423
)
2524
from twinkle.model.transformers.strategy import NativeFSDPStrategy

0 commit comments

Comments
 (0)