Skip to content

Commit b25b53d

Browse files
kleckistiepan
authored andcommitted
Fix Python Functions ops signature (#5143)
The parameters are not keyword-only as is the case for the dynamically generated APIs. Signed-off-by: Krzysztof Lecki <[email protected]>
1 parent f7d3da5 commit b25b53d

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

dali/python/nvidia/dali/ops/_operators/python_function.pyi

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -33,15 +33,15 @@ class PythonFunction:
3333
def __init__(
3434
self,
3535
/,
36-
*,
3736
function: Optional[Callable[..., Union[Any, Sequence[Any]]]] = None,
38-
batch_processing: bool = False,
3937
num_outputs: int = 1,
38+
device: str = "cpu",
39+
batch_processing: bool = False,
40+
*,
4041
output_layouts: Union[Sequence[str], str, None] = None,
4142
bytes_per_sample_hint: Union[Sequence[int], int, None] = [0],
4243
preserve: Optional[bool] = False,
4344
seed: Optional[int] = -1,
44-
device: Optional[str] = None,
4545
name: Optional[str] = None,
4646
) -> None:
4747
...
@@ -54,7 +54,7 @@ class PythonFunction:
5454
preserve: Optional[bool] = False,
5555
seed: Optional[int] = -1,
5656
name: Optional[str] = None,
57-
) -> Union[DataNode, Sequence[DataNode]]:
57+
) -> Union[DataNode, Sequence[DataNode], None]:
5858
"""See :meth:`nvidia.dali.ops.PythonFunction` class for complete information."""
5959
...
6060

@@ -76,16 +76,16 @@ class DLTensorPythonFunction:
7676
def __init__(
7777
self,
7878
/,
79-
*,
8079
function: Optional[Callable[..., Union[Any, Sequence[Any]]]] = None,
81-
batch_processing: bool = True,
8280
num_outputs: int = 1,
81+
device: str = "cpu",
82+
batch_processing: bool = True,
8383
synchronize_stream: Optional[bool] = True,
84+
*,
8485
output_layouts: Union[Sequence[str], str, None] = None,
8586
bytes_per_sample_hint: Union[Sequence[int], int, None] = [0],
8687
preserve: Optional[bool] = False,
8788
seed: Optional[int] = -1,
88-
device: Optional[str] = None,
8989
name: Optional[str] = None,
9090
) -> None:
9191
...
@@ -114,7 +114,7 @@ def python_function(
114114
seed: Optional[int] = -1,
115115
device: Optional[str] = None,
116116
name: Optional[str] = None,
117-
) -> Union[DataNode, Sequence[DataNode]]:
117+
) -> Union[DataNode, Sequence[DataNode], None]:
118118
"""
119119
Executes a Python function.
120120
@@ -140,7 +140,7 @@ def dl_tensor_python_function(
140140
seed: Optional[int] = -1,
141141
device: Optional[str] = None,
142142
name: Optional[str] = None,
143-
) -> Union[DataNode, Sequence[DataNode]]:
143+
) -> Union[DataNode, Sequence[DataNode], None]:
144144
"""
145145
Executes a Python function that operates on DLPack tensors.
146146

0 commit comments

Comments
 (0)