Linting

core/pipelines.py

@@ -141,14 +141,12 @@ def dense_passage_retrieval_ranker(
     ranker = SentenceTransformersRanker(model_name_or_path=ranker_model)
 
     search_pipeline.add_node(ranker, name="Ranker", inputs=["DPRRetriever"])
-    
+
     if audio_output:
         doc2speech = DocumentToSpeech(
             model_name_or_path="espnet/kan-bayashi_ljspeech_vits",
             generated_audio_dir=Path(data_path + "audio"),
         )
-        search_pipeline.add_node(
-            doc2speech, name="DocumentToSpeech", inputs=["Ranker"]
-        )
+        search_pipeline.add_node(doc2speech, name="DocumentToSpeech", inputs=["Ranker"])
 
     return search_pipeline, index_pipeline

interface/components.py

@@ -1,5 +1,10 @@
 import streamlit as st
-from interface.utils import get_pipelines, extract_text_from_url, extract_text_from_file, reset_vars_data
+from interface.utils import (
+    get_pipelines,
+    extract_text_from_url,
+    extract_text_from_file,
+    reset_vars_data,
+)
 from interface.draw_pipelines import get_pipeline_graph
 
 

interface/utils.py

@@ -25,12 +25,14 @@ def get_pipelines():
     ]
     return pipeline_names, pipeline_funcs, pipeline_func_parameters
 
+
 def reset_vars_data():
     st.session_state["doc_id"] = 0
     # Delete data files
     shutil.rmtree(data_path)
     os.makedirs(data_path, exist_ok=True)
 
+
 @st.experimental_memo
 def extract_text_from_url(url: str):
     article = Article(url)