Apply isort and black reformatting

Signed-off-by: gdengk <[email protected]>
NVIDIA · Jan 17, 2025 · 3d5accf · 3d5accf
1 parent 712de8f
commit 3d5accf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/core/optim/optimizer_with_main_params.py b/nemo/core/optim/optimizer_with_main_params.py
@@ -30,9 +30,9 @@
 
 try:
     from megatron.core.parallel_state import (
-        get_expert_data_parallel_group,
         get_data_parallel_group,
         get_data_parallel_world_size,
+        get_expert_data_parallel_group,
     )
     from megatron.core.tensor_parallel import copy_tensor_model_parallel_attributes