adc44ac9867933e9c237c18300fde5ed6c60a9be,elfi/client.py,BatchHandler,__init__,#BatchHandler#Any#Any#Any#,48

Before Change

    

    def __init__(self, model, output_names=None, client=None):
        self.client = client or get_client()
        self.compiled_net = self.client.compile(model.source_net, output_names)
        self.context = model.computation_context

        self._next_batch_index = 0
        self._pending_batches = OrderedDict()

After Change

    @property
    def pending_indices(self):
        return self._pending_batches.keys()

    def cancel_pending(self):
        Cancels all the pending batches and sets the next batch_index to the index of
        the last cancelled.

        Note that we rely here on the assumption that batches are processed in order.

Instances