style: apply ruff format to src/schemaforge/generators/drizzle_generator.py

Author: Coding-Dev-Tools

src/schemaforge/generators/drizzle_generator.py

@@ -1,4 +1,5 @@
 """Generator for Drizzle ORM TypeScript schema from SchemaForge IR."""
+
 from __future__ import annotations
 
 from ..ir import Column, ColumnType, EnumType, Schema, Table
@@ -26,7 +27,9 @@
 class DrizzleGenerator:
     """Generate Drizzle ORM TypeScript schema from Schema IR."""
 
-    def __init__(self, dialect: str = "pg", type_config: TypeConfig | None = None) -> None:
+    def __init__(
+        self, dialect: str = "pg", type_config: TypeConfig | None = None
+    ) -> None:
         """Initialize with dialect and optional custom type overrides.
 
         Args:
@@ -78,7 +81,11 @@ def _collect_imports(self, schema: Schema) -> str:
                 type_imports.add(drizzle_type)
 
                 # Add serial for primary keys with integer type
-                if col.primary_key and col.type == ColumnType.INTEGER and not col.default:
+                if (
+                    col.primary_key
+                    and col.type == ColumnType.INTEGER
+                    and not col.default
+                ):
                     type_imports.add("serial")
                     type_imports.discard("integer")