ruff lint fix

antithesis-tests/stress-composer/parallel_driver_delete.py

@@ -1,42 +1,42 @@
 #!/usr/bin/env -S python3 -u
 
 import json
+
 import limbo
-from utils import generate_random_value
 from antithesis.random import get_random
+from utils import generate_random_value
 
 # Get initial state
 try:
-    con_init = limbo.connect('init_state.db')
+    con_init = limbo.connect("init_state.db")
 except Exception as e:
     print(f"Error connecting to database: {e}")
     exit(0)
 
 cur_init = con_init.cursor()
 
-tbl_len = cur_init.execute('SELECT count FROM tables').fetchone()[0]
+tbl_len = cur_init.execute("SELECT count FROM tables").fetchone()[0]
 selected_tbl = get_random() % tbl_len
-tbl_schema = json.loads(cur_init.execute(f'SELECT schema FROM schemas WHERE tbl = {selected_tbl}').fetchone()[0])
+tbl_schema = json.loads(cur_init.execute(f"SELECT schema FROM schemas WHERE tbl = {selected_tbl}").fetchone()[0])
 
 # get primary key column
-pk = tbl_schema['pk']
+pk = tbl_schema["pk"]
 # get non-pk columns
-cols = [f'col_{col}' for col in range(tbl_schema['colCount']) if col != pk]
+cols = [f"col_{col}" for col in range(tbl_schema["colCount"]) if col != pk]
 
 try:
-    con = limbo.connect('stress_composer.db')
+    con = limbo.connect("stress_composer.db")
 except limbo.OperationalError as e:
-    print(f'Failed to open stress_composer.db. Exiting... {e}')
+    print(f"Failed to open stress_composer.db. Exiting... {e}")
     exit(0)
 cur = con.cursor()
 
 deletions = get_random() % 100
-print(f'Attempt to delete {deletions} rows in tbl_{selected_tbl}...')
+print(f"Attempt to delete {deletions} rows in tbl_{selected_tbl}...")
 
 for i in range(deletions):
     where_clause = f"col_{pk} = {generate_random_value(tbl_schema[f'col_{pk}']['data_type'])}"
 
-    cur.execute(f'''
+    cur.execute(f"""
         DELETE FROM tbl_{selected_tbl} WHERE {where_clause}
-    ''')
-
+    """)