Fix batch size propagation.

mzient · mzient · commit f23db794c6e7 · 2025-10-03T15:00:02.000+02:00
Signed-off-by: Michal Zientkiewicz &lt;michalz@nvidia.com&gt;
diff --git a/dali/python/nvidia/dali/experimental/dali2/_op_builder.py b/dali/python/nvidia/dali/experimental/dali2/_op_builder.py
@@ -475,7 +475,7 @@ def _infer_device():
             )
 
         # Call the operator (the result is an Invocation object)
-        return op_inst(*inputs, **call_args)
+        return op_inst(*inputs, batch_size=batch_size, **call_args)
 
     function = makefun.create_function(header, fn_call)
     function.op_class = op

Original file line number	Diff line number	Diff line change
`@@ -475,7 +475,7 @@ def _infer_device():`
`475`	`475`	`)`
`476`	`476`
`477`	`477`	`# Call the operator (the result is an Invocation object)`
`478`		`- return op_inst(inputs, *call_args)`
	`478`	`+ return op_inst(inputs, batch_size=batch_size, *call_args)`
`479`	`479`
`480`	`480`	`function = makefun.create_function(header, fn_call)`
`481`	`481`	`function.op_class = op`