class IteratorStore:
    def __init__(self, store):
        self._index = 0
        self.length = len(store)
        self.store = store

    def __next__(self):
        if self._index >= self.length:
            raise StopIteration
        else:
            item = (self.store[self._index].query_tensor,
                    self.store[self._index].response_tensor,
                    self.store[self._index].logprobs,
                    self.store[self._index].values,
                    self.store[self._index].rewards,
                    self.store[self._index].advantages,
                    self.store[self._index].returns,
                    self.store[self._index].pretrain_ids
                    )
            self._index += 1
            return item

    def __iter__(self):
        self._index = 0
        return self

    def __len__(self):
        return self.length