Skip to content

Commit fca97cb

Browse files
committed
fix: Apply ruff formatting
1 parent 1f88c23 commit fca97cb

2 files changed

Lines changed: 18 additions & 15 deletions

File tree

alembic/versions/20260310_67608331ee8a_add_parameter_nodes_table.py

Lines changed: 17 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
Create Date: 2026-03-10 18:29:54.555074
66
77
"""
8+
89
from typing import Sequence, Union
910

1011
import sqlalchemy as sa
@@ -13,26 +14,30 @@
1314
from alembic import op
1415

1516
# revision identifiers, used by Alembic.
16-
revision: str = '67608331ee8a'
17-
down_revision: Union[str, Sequence[str], None] = 'add_modelled_policies'
17+
revision: str = "67608331ee8a"
18+
down_revision: Union[str, Sequence[str], None] = "add_modelled_policies"
1819
branch_labels: Union[str, Sequence[str], None] = None
1920
depends_on: Union[str, Sequence[str], None] = None
2021

2122

2223
def upgrade() -> None:
2324
"""Upgrade schema."""
24-
op.create_table('parameter_nodes',
25-
sa.Column('name', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
26-
sa.Column('label', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
27-
sa.Column('description', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
28-
sa.Column('tax_benefit_model_version_id', sa.Uuid(), nullable=False),
29-
sa.Column('id', sa.Uuid(), nullable=False),
30-
sa.Column('created_at', sa.DateTime(), nullable=False),
31-
sa.ForeignKeyConstraint(['tax_benefit_model_version_id'], ['tax_benefit_model_versions.id'], ),
32-
sa.PrimaryKeyConstraint('id')
25+
op.create_table(
26+
"parameter_nodes",
27+
sa.Column("name", sqlmodel.sql.sqltypes.AutoString(), nullable=False),
28+
sa.Column("label", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
29+
sa.Column("description", sqlmodel.sql.sqltypes.AutoString(), nullable=True),
30+
sa.Column("tax_benefit_model_version_id", sa.Uuid(), nullable=False),
31+
sa.Column("id", sa.Uuid(), nullable=False),
32+
sa.Column("created_at", sa.DateTime(), nullable=False),
33+
sa.ForeignKeyConstraint(
34+
["tax_benefit_model_version_id"],
35+
["tax_benefit_model_versions.id"],
36+
),
37+
sa.PrimaryKeyConstraint("id"),
3338
)
3439

3540

3641
def downgrade() -> None:
3742
"""Downgrade schema."""
38-
op.drop_table('parameter_nodes')
43+
op.drop_table("parameter_nodes")

scripts/seed_models.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -356,9 +356,7 @@ def seed_model(
356356
nodes_to_add.append(node)
357357
seen_node_names.add(node.name)
358358

359-
console.print(
360-
f" Found {len(nodes_to_add)} parameter nodes (folder structure)"
361-
)
359+
console.print(f" Found {len(nodes_to_add)} parameter nodes (folder structure)")
362360

363361
with logfire.span("add_parameter_nodes", count=len(nodes_to_add)):
364362
node_rows = []

0 commit comments

Comments
 (0)