Skip to content

Commit 8abefa3

Browse files
Fix bug for passing kwargs to Batch UDF (#366)
* Fix bug for passing kwargs to Batch UDF * Use gobal defined skip list for batch udf kwargs
1 parent 3637e14 commit 8abefa3

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tiledb/cloud/dag/dag.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,8 @@
4949
_REPORT_TIMEOUT_SECS = 10
5050
"""The maximum request time when submitting non-essential log information."""
5151

52+
_SKIP_BATCH_UDF_KWARGS = ["image_name", "timeout", "result_format"]
53+
5254

5355
class ParentFailedError(futures.CancelledError):
5456
def __init__(self, cause: BaseException, node: "Node"):
@@ -1404,7 +1406,9 @@ def _build_batch_taskgraph(self):
14041406
args.append(models.TGUDFArgument(value=esc.visit(arg)))
14051407

14061408
for name, arg in node.kwargs.items():
1407-
if isinstance(arg, Node):
1409+
if name in _SKIP_BATCH_UDF_KWARGS:
1410+
continue
1411+
elif isinstance(arg, Node):
14081412
args.append(
14091413
models.TGUDFArgument(
14101414
name=name,

0 commit comments

Comments
 (0)