Updated logging error

summarizer/bert_parent.py

@@ -1,13 +1,10 @@
-import logging
 from typing import List
 
 import numpy as np
 import torch
 from numpy import ndarray
 from transformers import *
 
-logging.basicConfig(level=logging.WARNING)
-
 
 class BertParent(object):
 
@@ -28,8 +25,8 @@ class BertParent(object):
     def __init__(
         self,
         model: str,
-        custom_model: PreTrainedModel=None,
-        custom_tokenizer: PreTrainedTokenizer=None
+        custom_model: PreTrainedModel = None,
+        custom_tokenizer: PreTrainedTokenizer = None
     ):
         """
         :param model: Model is the string path for the bert weights. If given a keyword, the s3 path will be used
@@ -67,7 +64,7 @@ def tokenize_input(self, text: str) -> torch.tensor:
     def extract_embeddings(
         self,
         text: str,
-        hidden: int=-2,
+        hidden: int = -2,
         reduce_option: str ='mean'
     ) -> torch.Tensor:
 
@@ -100,7 +97,7 @@ def extract_embeddings(
     def create_matrix(
         self,
         content: List[str],
-        hidden: int=-2,
+        hidden: int = -2,
         reduce_option: str = 'mean'
     ) -> ndarray:
         """
@@ -120,7 +117,7 @@ def create_matrix(
     def __call__(
         self,
         content: List[str],
-        hidden: int= -2,
+        hidden: int = -2,
         reduce_option: str = 'mean'
     ) -> ndarray:
         return self.create_matrix(content, hidden, reduce_option)

summarizer/model_processors.py

@@ -274,7 +274,7 @@ class TransformerSummarizer(ModelProcessor):
         'TransfoXL': (TransfoXLModel, TransfoXLTokenizer),
         'XLNet': (XLNetModel, XLNetTokenizer),
         'XLM': (XLMModel, XLMTokenizer),
-        'DistilBert': (DistilBertModel, DistilBertTokenizer),
+        'DistilBert': (DistilBertModel, DistilBertTokenizer)
     }
 
     def __init__(