@@ -33,15 +33,15 @@ class PythonFunction:
33
33
def __init__ (
34
34
self ,
35
35
/ ,
36
- * ,
37
36
function : Optional [Callable [..., Union [Any , Sequence [Any ]]]] = None ,
38
- batch_processing : bool = False ,
39
37
num_outputs : int = 1 ,
38
+ device : str = "cpu" ,
39
+ batch_processing : bool = False ,
40
+ * ,
40
41
output_layouts : Union [Sequence [str ], str , None ] = None ,
41
42
bytes_per_sample_hint : Union [Sequence [int ], int , None ] = [0 ],
42
43
preserve : Optional [bool ] = False ,
43
44
seed : Optional [int ] = - 1 ,
44
- device : Optional [str ] = None ,
45
45
name : Optional [str ] = None ,
46
46
) -> None :
47
47
...
@@ -54,7 +54,7 @@ class PythonFunction:
54
54
preserve : Optional [bool ] = False ,
55
55
seed : Optional [int ] = - 1 ,
56
56
name : Optional [str ] = None ,
57
- ) -> Union [DataNode , Sequence [DataNode ]]:
57
+ ) -> Union [DataNode , Sequence [DataNode ], None ]:
58
58
"""See :meth:`nvidia.dali.ops.PythonFunction` class for complete information."""
59
59
...
60
60
@@ -76,16 +76,16 @@ class DLTensorPythonFunction:
76
76
def __init__ (
77
77
self ,
78
78
/ ,
79
- * ,
80
79
function : Optional [Callable [..., Union [Any , Sequence [Any ]]]] = None ,
81
- batch_processing : bool = True ,
82
80
num_outputs : int = 1 ,
81
+ device : str = "cpu" ,
82
+ batch_processing : bool = True ,
83
83
synchronize_stream : Optional [bool ] = True ,
84
+ * ,
84
85
output_layouts : Union [Sequence [str ], str , None ] = None ,
85
86
bytes_per_sample_hint : Union [Sequence [int ], int , None ] = [0 ],
86
87
preserve : Optional [bool ] = False ,
87
88
seed : Optional [int ] = - 1 ,
88
- device : Optional [str ] = None ,
89
89
name : Optional [str ] = None ,
90
90
) -> None :
91
91
...
@@ -114,7 +114,7 @@ def python_function(
114
114
seed : Optional [int ] = - 1 ,
115
115
device : Optional [str ] = None ,
116
116
name : Optional [str ] = None ,
117
- ) -> Union [DataNode , Sequence [DataNode ]]:
117
+ ) -> Union [DataNode , Sequence [DataNode ], None ]:
118
118
"""
119
119
Executes a Python function.
120
120
@@ -140,7 +140,7 @@ def dl_tensor_python_function(
140
140
seed : Optional [int ] = - 1 ,
141
141
device : Optional [str ] = None ,
142
142
name : Optional [str ] = None ,
143
- ) -> Union [DataNode , Sequence [DataNode ]]:
143
+ ) -> Union [DataNode , Sequence [DataNode ], None ]:
144
144
"""
145
145
Executes a Python function that operates on DLPack tensors.
146
146
0 commit comments