[AL-6294] | Bug Fix | Support for setting a 'from cursor for dataset.data_rows() (#1185)

Author: vbrodsky

labelbox/pagination.py

@@ -134,7 +134,7 @@ class _CursorPagination(_Pagination):
     def __init__(self, cursor_path: List[str], *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.cursor_path = cursor_path
-        self.next_cursor: Optional[Any] = None
+        self.next_cursor: Optional[Any] = kwargs.get('params', {}).get('from')
 
     def increment_page(self, results: Dict[str, Any]):
         for path in self.cursor_path:

tests/integration/test_data_rows.py

@@ -173,6 +173,9 @@ def test_data_row_bulk_creation(dataset, rand_gen, image_url):
     assert {data_row.row_data for data_row in data_rows} == {image_url}
     assert {data_row.global_key for data_row in data_rows} == {None}
 
+    data_rows = list(dataset.data_rows(from_cursor=data_rows[0].uid))
+    assert len(data_rows) == 1
+
     # Test creation using file name
     with NamedTemporaryFile() as fp:
         data = rand_gen(str).encode()