star

venv/lib/python3.9/site-packages/mlflow/store/entities/__init__.py

@@ -0,0 +1,3 @@
+from mlflow.store.entities.paged_list import PagedList
+
+__all__ = ["PagedList"]

venv/lib/python3.9/site-packages/mlflow/store/entities/paged_list.py

@@ -0,0 +1,18 @@
+from typing import 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)