Fix imports for LSG code: explicit BERT import + import math + logger

modeling_lsg_bert.py

@@ -1,10 +1,30 @@
 from logging import warn
-from transformers.models.bert.modeling_bert import *
+from transformers.models.bert.modeling_bert import (
+    BertPreTrainedModel,
+    BertEmbeddings,
+    BertModel,
+    BertForPreTraining,
+    BertForMaskedLM,
+    BertForNextSentencePrediction,
+    BertForSequenceClassification,
+    BertForMultipleChoice,
+    BertForTokenClassification,
+    BertForQuestionAnswering,
+    BertSelfOutput,
+    BertPooler,
+    BertEncoder,
+    BertLayer,
+)
 import torch
 import torch.nn as nn
 from transformers.models.bert.configuration_bert import BertConfig
+import math
 import sys
 
+import logging
+logger = logging.getLogger(__name__)
+
+
 AUTO_MAP = {
         "AutoModel": "modeling_lsg_bert.LSGBertModel",
         "AutoModelForCausalLM": "modeling_lsg_bert.LSGBertLMHeadModel",