Fix code style issues with Black

amosproj · Jul 18, 2023 · e1834f8 · e1834f8
1 parent d4b88e1
commit e1834f8
Show file tree

Hide file tree

Showing 5 changed files with 27 additions and 19 deletions.
diff --git a/QAChat/Common/deepL_translator.py b/QAChat/Common/deepL_translator.py
@@ -43,7 +43,11 @@ def __init__(self):
     def translate_to(self, text, target_lang, use_spacy_to_detect_lang_if_needed=True):
         if use_spacy_to_detect_lang_if_needed:
             doc = self.muulti_lang_nlp(text)
-            if doc._.language["language"] == "en" and doc._.language["score"] > 0.8 and target_lang == "EN-US":
+            if (
+                doc._.language["language"] == "en"
+                and doc._.language["score"] > 0.8
+                and target_lang == "EN-US"
+            ):
                 return Result(text, "EN_US")
 
         result = self.translator.translate_text(

diff --git a/QAChat/Data_Processing/data_preprocessor.py b/QAChat/Data_Processing/data_preprocessor.py
@@ -12,7 +12,7 @@
 class DataPreprocessor(ABC):
     @abstractmethod
     def load_preprocessed_data(
-            self, end_of_timeframe: datetime, start_of_timeframe: datetime
+        self, end_of_timeframe: datetime, start_of_timeframe: datetime
     ) -> List[DataInformation]:
         """
         Loads preprocessed data of a specific type that was created or modified

diff --git a/QAChat/Data_Processing/dummy_preprocessor.py b/QAChat/Data_Processing/dummy_preprocessor.py
@@ -13,7 +13,7 @@
 
 class DummyPreprocessor(DataPreprocessor):
     def load_preprocessed_data(
-            self, end_of_timeframe: datetime, start_of_timeframe: datetime
+        self, end_of_timeframe: datetime, start_of_timeframe: datetime
     ) -> List[DataInformation]:
         df = pd.read_csv("../../DummyData/qa_less_50.csv", sep=";")
         raw_data = []

diff --git a/QAChat/QA_Bot/qa_bot.py b/QAChat/QA_Bot/qa_bot.py
@@ -23,14 +23,14 @@
 
 class QABot:
     def __init__(
-            self,
-            embeddings=None,
-            database=None,
-            model=None,
-            translator=None,
-            embeddings_gpu=False,
-            repo_id="TheBloke/WizardLM-13B-V1-1-SuperHOT-8K-GGML",
-            filename="wizardlm-13b-v1.1-superhot-8k.ggmlv3.q5_0.bin",
+        self,
+        embeddings=None,
+        database=None,
+        model=None,
+        translator=None,
+        embeddings_gpu=False,
+        repo_id="TheBloke/WizardLM-13B-V1-1-SuperHOT-8K-GGML",
+        filename="wizardlm-13b-v1.1-superhot-8k.ggmlv3.q5_0.bin",
     ):
         self.answer = None
         self.context = None
@@ -61,11 +61,11 @@ def __init__(
             self.translator = DeepLTranslator()
 
     def get_llama_model(
-            self,
-            repo_id,
-            filename,
-            n_ctx=2048,
-            max_tokens=512,
+        self,
+        repo_id,
+        filename,
+        n_ctx=2048,
+        max_tokens=512,
     ):
         path = hf_hub_download(repo_id=repo_id, filename=filename)
 
@@ -81,7 +81,7 @@ def get_llama_model(
         )
 
     def answer_question_with_context(
-            self, question: str, context: List[str], handler=None
+        self, question: str, context: List[str], handler=None
     ) -> str:
         """
         This method takes a question and a list of context strings as input, and attempts to answer the question using the provided context.
@@ -150,7 +150,9 @@ def __sim_search(self, question: str) -> List[str]:
         ]
 
     def translate_text(self, question, language="EN-US"):
-        return self.translator.translate_to(question, language, use_spacy_to_detect_lang_if_needed=False)
+        return self.translator.translate_to(
+            question, language, use_spacy_to_detect_lang_if_needed=False
+        )
 
     def answer_question(self, question: str, handler: StreamLLMCallbackHandler | None):
         """

diff --git a/QAChat/QA_Bot/stream_LLM_callback_handler.py b/QAChat/QA_Bot/stream_LLM_callback_handler.py
@@ -22,5 +22,7 @@ def on_llm_new_token(self, token: str, **kwargs):
 
     def send_response(self, text):
         if self.lang != "EN-US":
-            text = self.translator.translate_to(text, self.lang, use_spacy_to_detect_lang_if_needed=False).text
+            text = self.translator.translate_to(
+                text, self.lang, use_spacy_to_detect_lang_if_needed=False
+            ).text
         return json.dumps({"text": text}) + "\n"