Method extend

rllib/env/utils/infinite_lookback_buffer.py:120–134

Appends all items in `items` to the end of this buffer.

(self, items)

            self.data.append(item)

    def extend(self, items) -> None:
        """Appends all items in `items` to the end of this buffer."""
        if self.finalized:
            # TODO (sven): When extending with a list of structs, we should
            #  probably rather do: `tree.map_structure(..., self.data,
            #  tree.map_structure(lambda *s: np.array(*s), *items)`)??
            self.data = tree.map_structure(
                lambda d, i: np.concatenate([d, i], axis=0),
                self.data,
                # Note, we could have dictionaries here.
                np.array(items) if isinstance(items, list) else items,
            )
        else:
            for item in items:
                self.append(item)

    def concat(self, other: "InfiniteLookbackBuffer") -> None:
        """Concatenates the data of `other` (w/o its lookback) to `self`.