Apply isort and black reformatting

Signed-off-by: akoumpa <akoumpa@users.noreply.github.com>

nemo/collections/llm/gpt/model/base.py

@@ -258,10 +258,10 @@ def configure_model(self) -> None:
         if not hasattr(self, "module"):
             if self.config.enable_cuda_graph:
                 assert HAVE_TE, "Transformer Engine is required for cudagraphs."
-                assert getattr(self.config,
-                    'use_te_rng_tracker', False
-                ), "Transformer engine's RNG tracker is required for cudagraphs, it can be "\
+                assert getattr(self.config, 'use_te_rng_tracker', False), (
+                    "Transformer engine's RNG tracker is required for cudagraphs, it can be "
                     "enabled with use_te_rng_tracker=True'."
+                )
 
             self.module = self.config.configure_model(self.tokenizer)
 

nemo/collections/nlp/models/language_modeling/megatron/gpt_full_te_layer_autocast_spec.py

@@ -15,8 +15,8 @@
 from importlib.metadata import version
 from typing import Any, Callable, Optional
 
-import torch
 import packaging
+import torch
 
 from nemo.collections.nlp.modules.common.megatron.utils import ApexGuardDefaults
 from nemo.collections.nlp.parts import utils_funcs