diff --git a/inference/core/workflows/execution_engine/entities/base.py b/inference/core/workflows/execution_engine/entities/base.py index 01657ec876..bc8ee40294 100644 --- a/inference/core/workflows/execution_engine/entities/base.py +++ b/inference/core/workflows/execution_engine/entities/base.py @@ -129,8 +129,7 @@ def init( "Attempted to initialise Batch object providing batch indices of size differing " "from size of the data." ) - - return cls(content=content, indices=indices) + return cls(content, indices) def __init__(self, content: List[B], indices: Optional[List[Tuple[int, ...]]]): self._content = content