diff options
-rw-r--r-- | alembic/op.pyi | 2 | ||||
-rw-r--r-- | alembic/operations/ops.py | 4 | ||||
-rw-r--r-- | alembic/operations/schemaobj.py | 2 |
3 files changed, 4 insertions, 4 deletions
diff --git a/alembic/op.pyi b/alembic/op.pyi index 2a03e1e..d781ac1 100644 --- a/alembic/op.pyi +++ b/alembic/op.pyi @@ -422,7 +422,7 @@ def bulk_insert( def create_check_constraint( constraint_name: Optional[str], table_name: str, - condition: "BinaryExpression", + condition: Union[str, "BinaryExpression"], schema: Optional[str] = None, **kw ) -> Optional["Table"]: diff --git a/alembic/operations/ops.py b/alembic/operations/ops.py index 7d69f09..99132dd 100644 --- a/alembic/operations/ops.py +++ b/alembic/operations/ops.py @@ -732,7 +732,7 @@ class CreateCheckConstraintOp(AddConstraintOp): self, constraint_name: Optional[str], table_name: str, - condition: Union["TextClause", "ColumnElement[Any]"], + condition: Union[str, "TextClause", "ColumnElement[Any]"], schema: Optional[str] = None, **kw ) -> None: @@ -778,7 +778,7 @@ class CreateCheckConstraintOp(AddConstraintOp): operations: "Operations", constraint_name: Optional[str], table_name: str, - condition: "BinaryExpression", + condition: Union[str, "BinaryExpression"], schema: Optional[str] = None, **kw ) -> Optional["Table"]: diff --git a/alembic/operations/schemaobj.py b/alembic/operations/schemaobj.py index 0d40dc7..3bff508 100644 --- a/alembic/operations/schemaobj.py +++ b/alembic/operations/schemaobj.py @@ -137,7 +137,7 @@ class SchemaObjects: self, name: Optional[str], source: str, - condition: Union["TextClause", "ColumnElement[Any]"], + condition: Union[str, "TextClause", "ColumnElement[Any]"], schema: Optional[str] = None, **kw ) -> Union["CheckConstraint"]: |