fixed formatting

Author: anngoroshi

test2text/pages/controls/controls_page.py

@@ -5,16 +5,19 @@ def controls_page():
     import streamlit as st
     import plotly.express as px
 
-
     st.header("Controls page")
     embedding_col, distances_col = st.columns(2)
     with embedding_col:
         st.subheader("Embedding")
 
         def refresh_counts():
             with get_db_client() as db:
-                st.session_state["all_annotations_count"] = db.count_all_entries_in_table("Annotations")
-                st.session_state["embedded_annotations_count"] = db.count_embedded_entries_in_table("Annotations")
+                st.session_state["all_annotations_count"] = (
+                    db.count_all_entries_in_table("Annotations")
+                )
+                st.session_state["embedded_annotations_count"] = (
+                    db.count_embedded_entries_in_table("Annotations")
+                )
 
         refresh_counts()
 

test2text/pages/reports/report_by_req.py

@@ -7,6 +7,7 @@
 
 SUMMARY_LENGTH = 100
 
+
 def make_a_report():
     from test2text.services.db import get_db_client
 
@@ -282,7 +283,6 @@ def write_annotations(current_annotations: set[tuple]):
                                                 "Annotations",
                                                 first_labels=[f"{req_external_id}"],
                                                 second_labels=anno_labels,
-
                                             )
                                         else:
                                             reqs_vectors_3d = minifold_vectors_3d(

test2text/pages/reports/report_by_tc.py

@@ -8,6 +8,7 @@
 
 SUMMARY_LENGTH = 100
 
+
 def make_a_tc_report():
     from test2text.services.db import get_db_client
 
@@ -32,8 +33,7 @@ def write_requirements(current_requirements: set[tuple]):
                 _,
                 distance,
             ) in current_requirements:
-                st.write(req_external_id, req_summary, round_distance(distance)
-                )
+                st.write(req_external_id, req_summary, round_distance(distance))
 
         with st.container(border=True):
             st.subheader("Filter test cases")
@@ -53,7 +53,6 @@ def write_requirements(current_requirements: set[tuple]):
             where_clauses = []
             params = []
 
-
             if filter_summary.strip():
                 where_clauses.append("Testcases.test_case LIKE ?")
                 params.append(f"%{filter_summary.strip()}%")

test2text/services/visualisation/visualize_vectors.py

@@ -103,7 +103,7 @@ def plot_2_sets_in_one_2d(
     first_color="red",
     second_color="green",
     first_labels=None,
-    second_labels=None
+    second_labels=None,
 ):
     fig = go.Figure()
     fig.add_trace(
@@ -142,7 +142,7 @@ def plot_2_sets_in_one_3d(
     first_color="red",
     second_color="green",
     first_labels=None,
-    second_labels=None
+    second_labels=None,
 ):
     fig = go.Figure()
     fig.add_trace(
@@ -154,7 +154,7 @@ def plot_2_sets_in_one_3d(
             name=first_title,
             text=first_labels,
             textposition="top left",
-            marker=dict(color=f"{first_color}")
+            marker=dict(color=f"{first_color}"),
         )
     )
 
@@ -167,7 +167,7 @@ def plot_2_sets_in_one_3d(
             name=second_title,
             text=second_labels,
             textposition="top center",
-            marker=dict(color=f"{second_color}")
+            marker=dict(color=f"{second_color}"),
         )
     )