[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
NVIDIA · Jan 26, 2024 · 4fcd37e · 4fcd37e
1 parent 92a9130
commit 4fcd37e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/nemo/collections/nlp/modules/common/megatron/megatron_init.py b/nemo/collections/nlp/modules/common/megatron/megatron_init.py
@@ -35,14 +35,13 @@
         get_pipeline_model_parallel_rank,
         set_expert_model_parallel_rank,
         set_expert_model_parallel_size,
+        set_expert_model_parallel_world_size,
         set_pipeline_model_parallel_rank,
         set_pipeline_model_parallel_split_rank,
         set_pipeline_model_parallel_world_size,
         set_tensor_model_parallel_rank,
         set_tensor_model_parallel_world_size,
         set_virtual_pipeline_model_parallel_rank,
-        set_expert_model_parallel_world_size,
-        set_expert_model_parallel_rank,
     )
 
 except (ImportError, ModuleNotFoundError):