From beb509650163508ab3cb30d423c221630421a108 Mon Sep 17 00:00:00 2001 From: Victarry Date: Tue, 26 Nov 2024 02:41:17 +0000 Subject: [PATCH] Apply isort and black reformatting Signed-off-by: Victarry --- nemo/core/optim/optimizer_with_main_params.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nemo/core/optim/optimizer_with_main_params.py b/nemo/core/optim/optimizer_with_main_params.py index ba2dd8ca80fd..19215440825e 100755 --- 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