Apply isort and black reformatting

Signed-off-by: suiyoubi <[email protected]>
NVIDIA · Dec 2, 2024 · c0a7836 · c0a7836
1 parent d363a11
commit c0a7836
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/llm/bert/model/base.py b/nemo/collections/llm/bert/model/base.py
@@ -48,7 +48,7 @@
     from megatron.core.models.bert.bert_model import BertModel as MCoreBert
 except (ImportError, ModuleNotFoundError) as e:
     HAVE_TE = False
-    MCoreBert = TransformerLayer # Place holder for import checking. BERT requires TE installed.
+    MCoreBert = TransformerLayer  # Place holder for import checking. BERT requires TE installed.
 
 if TYPE_CHECKING:
     from nemo.collections.common.tokenizers.tokenizer_spec import TokenizerSpec