Skip to content

batching

batching

Batchable

Batcher

Batcher(*args, **kwargs)

batched_args instance-attribute

batched_args = args

batched_kwargs instance-attribute

batched_kwargs = kwargs

last_batch_group instance-attribute

last_batch_group: Optional[List[int]] = None

needs_batching instance-attribute

needs_batching = False

current_value instance-attribute

current_value: Optional[Any] = None

total_batch_size property

total_batch_size

batch

batch(batchable: Batchable, *args, **kwargs) -> Tuple[Tuple[Any, Any], Optional[List[int]]]

narrow

narrow(batch_group: Optional[List[int]])

swap

swap(batch_group: Optional[List[int]], swap_value: Any)