Skip to content

Commit 1bfec18

Browse files
committed
Add missing decorator arg
1 parent 463f5d5 commit 1bfec18

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

singlestoredb/functions/decorator.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -142,6 +142,7 @@ def udf(
142142
name: Optional[str] = None,
143143
args: Optional[ParameterType] = None,
144144
returns: Optional[ReturnType] = None,
145+
timeout: Optional[int] = None,
145146
) -> Callable[..., Any]:
146147
"""
147148
Define a user-defined function (UDF).
@@ -169,6 +170,9 @@ def udf(
169170
Specifies the return data type of the function. This parameter
170171
works the same way as `args`. If the function is a table-valued
171172
function, the return type should be a `Table` object.
173+
timeout : int, optional
174+
The timeout in seconds for the UDF execution. If not specified,
175+
the global default timeout is used.
172176
173177
Returns
174178
-------
@@ -180,4 +184,5 @@ def udf(
180184
name=name,
181185
args=args,
182186
returns=returns,
187+
timeout=timeout,
183188
)

0 commit comments

Comments
 (0)