PR comments.

Author: anoto-moniz

src/citrine/_rest/pageable.py

@@ -1,4 +1,3 @@
-from functools import partial
 from typing import Optional, Iterable, Dict, Tuple, Callable, Union, Sequence
 from uuid import UUID
 
@@ -26,6 +25,8 @@ def _fetch_page(self,
                     per_page: Optional[int] = None,
                     json_body: Optional[dict] = None,
                     additional_params: Optional[dict] = None,
+                    *,
+                    version: Optional[str] = None
                     ) -> Tuple[Iterable[dict], str]:
         """
         Fetch visible elements.  This does not handle pagination.
@@ -59,6 +60,9 @@ def _fetch_page(self,
             }
         additional_params: dict, optional
             A dict that allows extra parameters to be added to the request parameters
+        version: str, optional
+            A string denoting which version of the underlying API endpoint will be called. Defaults
+            to the collection's API version.
 
         Returns
         -------
@@ -69,15 +73,16 @@ def _fetch_page(self,
 
         """
         # To avoid setting defaults -> reduce mutation risk, and to make more extensible
-        path = self._get_path() if path is None else path
-        fetch_func = fetch_func or partial(self.session.get_resource, version=self._api_version)
-        json_body = {} if json_body is None else json_body
+        path = path or self._get_path()
+        fetch_func = fetch_func or self.session.get_resource
+        json_body = json_body or {}
 
-        module_type = getattr(self, '_module_type', None)
-        params = self._page_params(page, per_page, module_type)
+        params = self._page_params(page, per_page)
         params.update(additional_params or {})
 
-        data = fetch_func(path, params=params, **json_body)
+        version = version or self._api_version
+
+        data = fetch_func(path, params=params, version=version, **json_body)
 
         try:
             next_uri = data.get('next', "")

src/citrine/resources/predictor.py

@@ -373,8 +373,8 @@ def _list_base(self, *, per_page: int = 100, archived: Optional[bool] = None):
             filters["archived"] = archived
 
         fetcher = partial(self._fetch_page,
-                          fetch_func=partial(self.session.get_resource, version="v4"),
-                          additional_params=filters)
+                          additional_params=filters,
+                          version="v4")
         return self._paginator.paginate(page_fetcher=fetcher,
                                         collection_builder=self._build_collection_elements,
                                         per_page=per_page)