🛠️ apply pre-commit fixes

Author: getsantry[bot]

src/migrations/versions/b2a0463d9b91_migration.py

@@ -5,42 +5,51 @@
 Create Date: 2025-03-03 17:34:37.242326
 
 """
-from alembic import op
-import sqlalchemy as sa
 
+import sqlalchemy as sa
+from alembic import op
 
 # revision identifiers, used by Alembic.
-revision = 'b2a0463d9b91'
-down_revision = 'e0fcdc14251c'
+revision = "b2a0463d9b91"
+down_revision = "e0fcdc14251c"
 branch_labels = None
 depends_on = None
 
 
 def upgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    op.create_table('codegen_unit_test_generation_pr_context_to_run_id',
-    sa.Column('id', sa.Integer(), nullable=False),
-    sa.Column('provider', sa.String(), nullable=False),
-    sa.Column('owner', sa.String(), nullable=False),
-    sa.Column('pr_id', sa.BigInteger(), nullable=False),
-    sa.Column('repo', sa.String(), nullable=False),
-    sa.Column('run_id', sa.Integer(), nullable=False),
-    sa.Column('iterations', sa.Integer(), nullable=False),
-    sa.Column('original_pr_url', sa.String(), nullable=False),
-    sa.ForeignKeyConstraint(['run_id'], ['run_state.id'], ondelete='CASCADE'),
-    sa.PrimaryKeyConstraint('id'),
-    sa.UniqueConstraint('provider', 'pr_id', 'repo', 'owner', 'original_pr_url')
+    op.create_table(
+        "codegen_unit_test_generation_pr_context_to_run_id",
+        sa.Column("id", sa.Integer(), nullable=False),
+        sa.Column("provider", sa.String(), nullable=False),
+        sa.Column("owner", sa.String(), nullable=False),
+        sa.Column("pr_id", sa.BigInteger(), nullable=False),
+        sa.Column("repo", sa.String(), nullable=False),
+        sa.Column("run_id", sa.Integer(), nullable=False),
+        sa.Column("iterations", sa.Integer(), nullable=False),
+        sa.Column("original_pr_url", sa.String(), nullable=False),
+        sa.ForeignKeyConstraint(["run_id"], ["run_state.id"], ondelete="CASCADE"),
+        sa.PrimaryKeyConstraint("id"),
+        sa.UniqueConstraint("provider", "pr_id", "repo", "owner", "original_pr_url"),
     )
-    with op.batch_alter_table('codegen_unit_test_generation_pr_context_to_run_id', schema=None) as batch_op:
-        batch_op.create_index('ix_autofix_repo_owner_pr_id', ['owner', 'repo', 'pr_id', 'original_pr_url'], unique=False)
+    with op.batch_alter_table(
+        "codegen_unit_test_generation_pr_context_to_run_id", schema=None
+    ) as batch_op:
+        batch_op.create_index(
+            "ix_autofix_repo_owner_pr_id",
+            ["owner", "repo", "pr_id", "original_pr_url"],
+            unique=False,
+        )
 
     # ### end Alembic commands ###
 
 
 def downgrade():
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('codegen_unit_test_generation_pr_context_to_run_id', schema=None) as batch_op:
-        batch_op.drop_index('ix_autofix_repo_owner_pr_id')
+    with op.batch_alter_table(
+        "codegen_unit_test_generation_pr_context_to_run_id", schema=None
+    ) as batch_op:
+        batch_op.drop_index("ix_autofix_repo_owner_pr_id")
 
-    op.drop_table('codegen_unit_test_generation_pr_context_to_run_id')
+    op.drop_table("codegen_unit_test_generation_pr_context_to_run_id")
     # ### end Alembic commands ###

src/seer/automation/codegen/retry_unittest_step.py

@@ -8,7 +8,6 @@
     AUTOFIX_EXECUTION_HARD_TIME_LIMIT_SECS,
     AUTOFIX_EXECUTION_SOFT_TIME_LIMIT_SECS,
 )
-
 from seer.automation.codebase.repo_client import RepoClientType
 from seer.automation.codegen.models import CodeUnitTestRequest
 from seer.automation.codegen.retry_unittest_coding_component import RetryUnitTestCodingComponent

src/seer/automation/codegen/unit_test_github_pr_creator.py

@@ -5,10 +5,7 @@
 
 from seer.automation.codebase.repo_client import RepoClient
 from seer.automation.models import FileChange
-from seer.db import (
-    DbPrContextToUnitTestGenerationRunIdMapping,
-    Session,
-)
+from seer.db import DbPrContextToUnitTestGenerationRunIdMapping, Session
 
 logger = logging.getLogger(__name__)