fixed formatting

Author: anngoroshi

test2text/pages/controls/controls_page.py

@@ -12,11 +12,11 @@ def controls_page():
 
         def refresh_counts():
             with get_db_client() as db:
-                st.session_state["all_annotations_count"] = (
-                    db.count_all_entries("Annotations")
+                st.session_state["all_annotations_count"] = db.count_all_entries(
+                    "Annotations"
                 )
                 st.session_state["embedded_annotations_count"] = (
-                    db.count_notnull_entries("embedding",from_table="Annotations")
+                    db.count_notnull_entries("embedding", from_table="Annotations")
                 )
 
         refresh_counts()

test2text/pages/reports/report_by_tc.py

@@ -80,9 +80,7 @@ def write_requirements(current_requirements: set[tuple]):
                     for (tc_id, _, test_case, distance) in data
                 }
             else:
-                tc_dict = {
-                    test_case: tc_id for (tc_id, _, test_case) in data
-                }
+                tc_dict = {test_case: tc_id for (tc_id, _, test_case) in data}
 
             st.subheader("Choose ONE of filtered test cases")
             option = st.selectbox(

test2text/services/db/client.py

@@ -116,9 +116,7 @@ def count_all_entries(self, from_table: str) -> int:
         count = self.conn.execute(f"SELECT COUNT(*) FROM {from_table}").fetchone()[0]
         return count
 
-    def count_notnull_entries(
-        self, *columns: str, from_table: str
-    ) -> int:
+    def count_notnull_entries(self, *columns: str, from_table: str) -> int:
         count = self.conn.execute(
             f"SELECT COUNT(*) FROM {from_table} WHERE {' AND  '.join([column + ' IS NOT NULL' for column in columns])}"
         ).fetchone()[0]

test2text/services/embeddings/annotation_embeddings_controls.py

@@ -13,7 +13,9 @@ def embed_annotations(*_, embed_all=False, on_progress: OnProgress = None):
         from .embed import embed_annotations_batch
 
         annotations_count = db.count_all_entries("Annotations")
-        embedded_annotations_count = db.count_notnull_entries("embedding",from_table="Annotations")
+        embedded_annotations_count = db.count_notnull_entries(
+            "embedding", from_table="Annotations"
+        )
         if embed_all:
             annotations_to_embed = annotations_count
         else: