From 4c0730aa62ed396c106e77509e6f49b10f4e4448 Mon Sep 17 00:00:00 2001 From: Nick Crews Date: Wed, 10 May 2023 10:53:05 -0800 Subject: [PATCH] fixup: mypy --- lib/sqlalchemy/sql/_elements_constructors.py | 7 +++++-- lib/sqlalchemy/sql/elements.py | 2 +- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/lib/sqlalchemy/sql/_elements_constructors.py b/lib/sqlalchemy/sql/_elements_constructors.py index cdbe7be7d6..10d43a0698 100644 --- a/lib/sqlalchemy/sql/_elements_constructors.py +++ b/lib/sqlalchemy/sql/_elements_constructors.py @@ -907,7 +907,10 @@ def cast( return Cast(expression, type_) -def try_cast(*arg, **kw): +def try_cast( + expression: _ColumnExpressionOrLiteralArgument[Any], + type_: _TypeEngineArgument[_T], + ) -> TryCast[_T]: """Produce a ``TRY_CAST`` expression. :func:`.try_cast` returns an instance of :class:`.TryCast`. @@ -938,7 +941,7 @@ def try_cast(*arg, **kw): .. versionadded:: 2.0.14 """ - return TryCast(*arg, **kw) + return TryCast(expression, type_) def column( diff --git a/lib/sqlalchemy/sql/elements.py b/lib/sqlalchemy/sql/elements.py index 66081d1b23..884e2b90fe 100644 --- a/lib/sqlalchemy/sql/elements.py +++ b/lib/sqlalchemy/sql/elements.py @@ -3414,7 +3414,7 @@ class Cast(WrapsColumnExpression[_T]): return self.clause -class TryCast(Cast): +class TryCast(Cast[_T]): """Represent a TRY_CAST expression. Details on :class:`.TryCast` usage is at :func:`.try_cast`. -- 2.47.3