1 files changed, 2 insertions, 7 deletions
diff --git a/shared/tests/test_sa_models.py b/shared/tests/test_sa_models.py
index de994c5..67c3c82 100644
--- a/shared/tests/test_sa_models.py
+++ b/shared/tests/test_sa_models.py
@@ -1,6 +1,5 @@
 """Tests for SQLAlchemy ORM models."""
 
-import pytest
 from sqlalchemy import inspect
 
 
@@ -117,9 +116,7 @@ class TestOrderRow:
         from shared.sa_models import OrderRow
 
         table = OrderRow.__table__
-        fk_cols = {
-            fk.parent.name: fk.target_fullname for fk in table.foreign_keys
-        }
+        fk_cols = {fk.parent.name: fk.target_fullname for fk in table.foreign_keys}
         assert fk_cols == {"signal_id": "signals.id"}
 
 
@@ -157,9 +154,7 @@ class TestTradeRow:
         from shared.sa_models import TradeRow
 
         table = TradeRow.__table__
-        fk_cols = {
-            fk.parent.name: fk.target_fullname for fk in table.foreign_keys
-        }
+        fk_cols = {fk.parent.name: fk.target_fullname for fk in table.foreign_keys}
         assert fk_cols == {"order_id": "orders.id"}