Skip to content

Commit 5493ed8

Browse files
committed
Fix lint error
1 parent ac20fd2 commit 5493ed8

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

tests/unit_tests/context_parallel/test_packing_and_hybrid_cp.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -11,15 +11,11 @@
1111

1212
from megatron.core import mpu, parallel_state
1313
from megatron.core.datasets.data_schedule import get_batch_on_this_rank_for_sequence_packing
14-
from megatron.core.models.gpt.gpt_layer_specs import (
15-
get_gpt_decoder_block_spec,
16-
)
14+
from megatron.core.models.gpt.gpt_layer_specs import get_gpt_decoder_block_spec
1715
from megatron.core.models.gpt.gpt_layer_specs import (
1816
get_gpt_layer_with_transformer_engine_spec as gpt_te_spec,
1917
)
20-
from megatron.core.models.gpt.gpt_layer_specs import (
21-
get_gpt_mtp_block_spec,
22-
)
18+
from megatron.core.models.gpt.gpt_layer_specs import (get_gpt_mtp_block_spec
2319
from megatron.core.models.gpt.gpt_model import GPTModel
2420
from megatron.core.num_microbatches_calculator import (
2521
init_num_microbatches_calculator,

0 commit comments

Comments
 (0)