Fix type annotations in create_foreign_key (#1430)

The constraint name parameter of create_foreign_key should be optional, but the batch function defined it as str instead of Optional[str]. Closes #1429
sqlalchemy · Feb 22, 2024 · 88769c5 · 88769c5
1 parent a22d21a
commit 88769c5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/alembic/operations/base.py b/alembic/operations/base.py
@@ -1721,7 +1721,7 @@ def create_exclude_constraint(
 
         def create_foreign_key(
             self,
-            constraint_name: str,
+            constraint_name: Optional[str],
             referent_table: str,
             local_cols: List[str],
             remote_cols: List[str],

diff --git a/alembic/operations/ops.py b/alembic/operations/ops.py
@@ -681,7 +681,7 @@ def create_foreign_key(
     def batch_create_foreign_key(
         cls,
         operations: BatchOperations,
-        constraint_name: str,
+        constraint_name: Optional[str],
         referent_table: str,
         local_cols: List[str],
         remote_cols: List[str],