diff options
Diffstat (limited to 'alembic/operations')
| -rw-r--r-- | alembic/operations/ops.py | 4 | ||||
| -rw-r--r-- | alembic/operations/schemaobj.py | 2 |
2 files changed, 3 insertions, 3 deletions
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"]: |