diff options
Diffstat (limited to 'lib/sqlalchemy/exc.py')
| -rw-r--r-- | lib/sqlalchemy/exc.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/lib/sqlalchemy/exc.py b/lib/sqlalchemy/exc.py index cc78e0971..8f4b963eb 100644 --- a/lib/sqlalchemy/exc.py +++ b/lib/sqlalchemy/exc.py @@ -33,6 +33,7 @@ if typing.TYPE_CHECKING: from .engine.interfaces import _DBAPIAnyExecuteParams from .engine.interfaces import Dialect from .sql.compiler import Compiled + from .sql.compiler import TypeCompiler from .sql.elements import ClauseElement if typing.TYPE_CHECKING: @@ -221,8 +222,8 @@ class UnsupportedCompilationError(CompileError): def __init__( self, - compiler: "Compiled", - element_type: Type["ClauseElement"], + compiler: Union[Compiled, TypeCompiler], + element_type: Type[ClauseElement], message: Optional[str] = None, ): super(UnsupportedCompilationError, self).__init__( |