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

for more information, see https://pre-commit.ci

nemo/core/optim/distributed_adam.py

@@ -16,20 +16,18 @@
 from typing import Callable, Dict, Iterable, Optional, Union
 
 import torch
-
+from apex.contrib.optimizers.distributed_fused_adam import (
+    DistributedFusedAdam,
+    _disable_pre_forward_hook,
+    _multi_tensor_copy,
+)
 from megatron.core import parallel_state
 from megatron.core.dist_checkpointing.dict_utils import dict_list_map_inplace
 from megatron.core.dist_checkpointing.mapping import ShardedTensor
 from megatron.core.dist_checkpointing.optimizer import get_param_id_to_sharded_param_map, optim_state_to_sharding_state
 
 from nemo.utils import str_to_dtype
 
-from apex.contrib.optimizers.distributed_fused_adam import (
-    DistributedFusedAdam,
-    _disable_pre_forward_hook,
-    _multi_tensor_copy,
-)
-
 # Check if Transformer Engine has FP8 tensor class
 HAVE_TE_FP8TENSOR = False
 try: