fix linter

dashboards/pages/01_🔍_Search_Judgements.py

@@ -38,9 +38,7 @@ def get_embedding_model() -> Any:
 
 with st.form(key="search_form"):
     query = st.text_area("What you are looking for in the judgements?")
-    max_judgements = st.slider(
-        "Max judgements to show", min_value=1, max_value=20, value=5
-    )
+    max_judgements = st.slider("Max judgements to show", min_value=1, max_value=20, value=5)
     submit_button = st.form_submit_button(label="Search")
 
 if submit_button:

juddges/data/database.py

@@ -62,9 +62,7 @@ def update_func (document: dict[str, Any]) -> dict[str, Any]:
     - Update is called specified documents.
     """
 
-    def __init__(
-        self, mongo_uri: str, update_func: Callable[[dict[str, Any]], dict]
-    ) -> None:
+    def __init__(self, mongo_uri: str, update_func: Callable[[dict[str, Any]], dict]) -> None:
         self.mongo_uri = mongo_uri
         self.update_func = update_func
 

juddges/retrieval/mongo_hybrid_search.py

@@ -8,7 +8,7 @@ def run_hybrid_search(
     text_priority: float = 1,
 ):
     num_candidates = limit * 10
-    
+
     vector_search = {
         "$vectorSearch": {
             "index": "vector_index",
@@ -25,9 +25,7 @@ def run_hybrid_search(
 
     def make_compute_score_doc(priority, score_field_name):
         return {
-            "$addFields": {
-                score_field_name: {"$divide": [1.0, {"$add": ["$rank", priority, 1]}]}
-            }
+            "$addFields": {score_field_name: {"$divide": [1.0, {"$add": ["$rank", priority, 1]}]}}
         }
 
     def make_projection_doc(score_field_name):

scripts/sft/fine_tune_llm.py

@@ -2,6 +2,7 @@
 Fine-tune a large language model using SFT.
 the script is based on: https://www.philschmid.de/fine-tune-llms-in-2024-with-trl
 """
+
 import os
 from pathlib import Path