Fixes linting

core/pipelines.py

@@ -11,9 +11,10 @@ from haystack.nodes.ranker import SentenceTransformersRanker
 from haystack.nodes.audio.document_to_speech import DocumentToSpeech
 import os
 
-data_path = 'data/'
+data_path = "data/"
 os.makedirs(data_path, exist_ok=True)
 
+
 def keyword_search(index="documents", split_word_length=100, audio_output=False):
     """
     **Keyword Search Pipeline**
@@ -49,13 +50,15 @@ def keyword_search(index="documents", split_word_length=100, audio_output=False)
     index_pipeline.add_node(
         document_store, name="DocumentStore", inputs=["TfidfRetriever"]
     )
-    
+
     if audio_output:
         doc2speech = DocumentToSpeech(
             model_name_or_path="espnet/kan-bayashi_ljspeech_vits",
-            generated_audio_dir=Path(data_path + 'audio'),
+            generated_audio_dir=Path(data_path + "audio"),
+        )
+        search_pipeline.add_node(
+            doc2speech, name="DocumentToSpeech", inputs=["TfidfRetriever"]
         )
-        search_pipeline.add_node(doc2speech, name='DocumentToSpeech', inputs=['TfidfRetriever'])
 
     return search_pipeline, index_pipeline
 

core/search_index.py

@@ -38,12 +38,10 @@ def search(queries, pipeline):
                 "fragment_id": res.id,
             }
             if not score_is_empty:
-                match.update({'score': res.score})
-            if hasattr(res, 'content_audio'):
-                match.update({'content_audio': res.content_audio})
-            query_results.append(
-                match
-            )
+                match.update({"score": res.score})
+            if hasattr(res, "content_audio"):
+                match.update({"content_audio": res.content_audio})
+            query_results.append(match)
         if not score_is_empty:
             query_results = sorted(
                 query_results, key=lambda x: x["score"], reverse=True

interface/components.py

@@ -59,10 +59,10 @@ def component_show_search_result(container, results):
             st.markdown(f"### Match {idx+1}")
             st.markdown(f"**Text**: {document['text']}")
             st.markdown(f"**Document**: {document['id']}")
-            if 'score' in document:
+            if "score" in document:
                 st.markdown(f"**Score**: {document['score']:.3f}")
-            if 'content_audio' in document:
-                st.audio(str(document['content_audio']))
+            if "content_audio" in document:
+                st.audio(str(document["content_audio"]))
             st.markdown("---")