Source code for mlflow.store.entities.paged_list

from typing import List, TypeVar

T = TypeVar("T")


class PagedList(List[T]):
    """
    Wrapper class around the base Python `List` type. Contains an additional `token`  string
    attribute that can be passed to the pagination API that returned this list to fetch additional
    elements, if any are available
    """

    def __init__(self, items: List[T], token):
        super().__init__(items)
        self.token = token

    def to_list(self):
        return list(self)