Apply isort and black reformatting

Signed-off-by: akoumpa <[email protected]>
NVIDIA · Sep 27, 2024 · d745aff · d745aff
1 parent 42b5244
commit d745aff
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/nemo/collections/nlp/modules/common/huggingface/__init__.py b/nemo/collections/nlp/modules/common/huggingface/__init__.py
@@ -20,4 +20,4 @@
     get_huggingface_lm_model,
     get_huggingface_pretrained_lm_models_list,
 )
-from nemo.collections.nlp.modules.common.huggingface.roberta import RobertaEncoder
+from nemo.collections.nlp.modules.common.huggingface.roberta import RobertaEncoder
diff --git a/nemo/collections/nlp/modules/common/huggingface/huggingface_utils.py b/nemo/collections/nlp/modules/common/huggingface/huggingface_utils.py
@@ -25,14 +25,13 @@
     RobertaConfig,
 )
 
-from nemo.utils import logging
-
 from nemo.collections.nlp.modules.common.huggingface.albert import AlbertEncoder
 from nemo.collections.nlp.modules.common.huggingface.bert import BertEncoder
 from nemo.collections.nlp.modules.common.huggingface.camembert import CamembertEncoder
 from nemo.collections.nlp.modules.common.huggingface.distilbert import DistilBertEncoder
 from nemo.collections.nlp.modules.common.huggingface.gpt2 import GPT2Encoder
 from nemo.collections.nlp.modules.common.huggingface.roberta import RobertaEncoder
+from nemo.utils import logging
 
 __all__ = ["get_huggingface_lm_model", "get_huggingface_pretrained_lm_models_list", "VOCAB_FILE_NAME"]
 

diff --git a/nemo/collections/nlp/modules/common/lm_utils.py b/nemo/collections/nlp/modules/common/lm_utils.py
@@ -20,8 +20,6 @@
 from omegaconf import DictConfig
 from pytorch_lightning import Trainer
 
-from nemo.utils import AppState, logging
-
 from nemo.collections.nlp.modules.common.bert_module import BertModule
 from nemo.collections.nlp.modules.common.decoder_module import DecoderModule
 from nemo.collections.nlp.modules.common.encoder_module import EncoderModule
@@ -35,6 +33,7 @@
     get_huggingface_transformer,
     get_nemo_transformer,
 )
+from nemo.utils import AppState, logging
 
 __all__ = ['get_pretrained_lm_models_list', 'get_lm_model', 'pad_batch']