chore: tidy up

Author: Daniel Bush

sypht/client.py

@@ -421,7 +421,7 @@ def get_annotations(
     ):
         page_iter = fetch_all_pages(
             name="get_annotations",
-            fetch_page=lambda *args, **kwargs: self._get_annotations(*args, **kwargs),
+            fetch_page=self._get_annotations,
             get_page=lambda response: response["annotations"],
         )
         annotations = []
@@ -474,9 +474,7 @@ def _get_annotations(
     def get_annotations_for_docs(self, doc_ids, endpoint=None):
         page_iter = fetch_all_pages(
             name="get_annotations_for_docs",
-            fetch_page=lambda *args, **kwargs: self._get_annotations_for_docs(
-                *args, **kwargs
-            ),
+            fetch_page=self._get_annotations_for_docs,
             get_page=lambda response: response["annotations"],
         )
         annotations = []

sypht/util.py

@@ -9,7 +9,7 @@ def fetch_all_pages(
 ) -> Callable[..., Iterator[Any]]:
     """Returns an iterator that calls fetch_page with an offset that we increment by the number of pages fetched.  Stop if page returns empty list.
 
-    :param fetch_page: the api call that takes an offset (zero-based) that fetches a page of results
+    :param fetch_page: a function that makes an api call to fetch a page of results (using zero-based offset)
     :param get_page: a function that extracts the page from the response
     """