update package import

Signed-off-by: Yang Zhang <yangzhang@nvidia.com>
NVIDIA · Feb 7, 2020 · 2909fcf · 2909fcf
1 parent 1110589
commit 2909fcf
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 6 deletions.
diff --git a/examples/nlp/asr_postprocessor/asr_postprocessor.py b/examples/nlp/asr_postprocessor/asr_postprocessor.py
@@ -19,7 +19,6 @@
 
 import torch
 
-import nemo
 import nemo.collections.nlp as nemo_nlp
 import nemo.collections.nlp.nm.data_layers.machine_translation_datalayer
 from nemo import logging

diff --git a/examples/nlp/intent_detection_slot_tagging/joint_intent_slot_infer.py b/examples/nlp/intent_detection_slot_tagging/joint_intent_slot_infer.py
@@ -51,7 +51,7 @@
 See the list of pretrained models, call:
 nemo_nlp.huggingface.BERT.list_pretrained_models()
 """
-pretrained_bert_model = nemo.collections.nlp.nm.trainables.common.huggingface.BERT(
+pretrained_bert_model = nemo.collections.nlp.nm.trainables.huggingface.BERT(
     pretrained_model_name=args.pretrained_bert_model
 )
 hidden_size = pretrained_bert_model.hidden_size

diff --git a/examples/nlp/intent_detection_slot_tagging/joint_intent_slot_with_bert.py b/examples/nlp/intent_detection_slot_tagging/joint_intent_slot_with_bert.py
@@ -87,12 +87,12 @@
 nemo_nlp.huggingface.BERT.list_pretrained_models()
 """
 if args.bert_checkpoint and args.bert_config:
-    pretrained_bert_model = nemo.collections.nlp.nm.trainables.common.huggingface.BERT(
+    pretrained_bert_model = nemo.collections.nlp.nm.trainables.huggingface.BERT(
         config_filename=args.bert_config
     )
     pretrained_bert_model.restore_from(args.bert_checkpoint)
 else:
-    pretrained_bert_model = nemo.collections.nlp.nm.trainables.common.huggingface.BERT(
+    pretrained_bert_model = nemo.collections.nlp.nm.trainables.huggingface.BERT(
         pretrained_model_name=args.pretrained_bert_model
     )
 

diff --git a/examples/nlp/question_answering/question_answering_squad.py b/examples/nlp/question_answering/question_answering_squad.py
@@ -66,7 +66,6 @@
 import json
 import os
 
-import nemo
 import nemo.collections.nlp as nemo_nlp
 import nemo.core as nemo_core
 from nemo import logging

diff --git a/nemo/collections/nlp/data/datasets/qa_squad_dataset.py b/nemo/collections/nlp/data/datasets/qa_squad_dataset.py
@@ -19,7 +19,6 @@
 import json
 import os
 import pickle
-import sys
 
 import numpy as np
 import torch