openprotein-python 0.14.0__tar.gz → 0.14.1__tar.gz

{openprotein_python-0.14.0 → openprotein_python-0.14.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openprotein-python
-Version: 0.14.0
+Version: 0.14.1
 Summary: OpenProtein Python interface.
 Author-email: Mark Gee <markgee@ne47.bio>, "Timothy Truong Jr." <ttruong@ne47.bio>, Tristan Bepler <tbepler@ne47.bio>
 License-Expression: MIT

{openprotein_python-0.14.0 → openprotein_python-0.14.1}/openprotein/clustering/api.py

@@ -17,17 +17,17 @@ PATH_PREFIX = "v1/clustering"
 def clustering_list_get(
     session: APISession,
     method: str | None = None,
-    page_size: int | None = None,
-    page_offset: int | None = None,
+    limit: int | None = None,
+    offset: int | None = None,
 ) -> list[ClusteringMetadata]:
     """List clustering jobs, optionally filtered by method."""
     params: dict = {}
     if method is not None:
         params["method"] = method
-    if page_size is not None:
-        params["page_size"] = page_size
-    if page_offset is not None:
-        params["page_offset"] = page_offset
+    if limit is not None:
+        params["limit"] = limit
+    if offset is not None:
+        params["offset"] = offset
     response = session.get(PATH_PREFIX, params=params or None)
     return TypeAdapter(list[ClusteringMetadata]).validate_python(response.json())
 

{openprotein_python-0.14.0 → openprotein_python-0.14.1}/openprotein/clustering/clustering.py

@@ -64,9 +64,7 @@ class ClusteringAPI:
         assay_id = (
             assay.assay_id
             if isinstance(assay, AssayMetadata)
-            else assay.id
-            if isinstance(assay, AssayDataset)
-            else assay
+            else assay.id if isinstance(assay, AssayDataset) else assay
         )
         if sequences is not None and assay_id is not None:
             raise InvalidParameterError(
@@ -86,9 +84,7 @@ class ClusteringAPI:
         feature_type = (
             FeatureType.PLM
             if isinstance(model, EmbeddingModel)
-            else FeatureType.SVD
-            if isinstance(model, SVDModel)
-            else feature_type
+            else FeatureType.SVD if isinstance(model, SVDModel) else feature_type
         )
         if feature_type is None:
             raise InvalidParameterError(
@@ -175,12 +171,13 @@ class ClusteringAPI:
     def list(
         self,
         method: str | None = None,
-        page_size: int | None = None,
-        page_offset: int | None = None,
+        limit: int | None = None,
+        offset: int | None = None,
     ) -> list[HierarchicalClusteringFuture]:
         """List clustering jobs, optionally filtering by method.
 
-        Pass `page_size` / `page_offset` to paginate."""
+        Pass `limit` / `offset` to paginate (the clustering list endpoint
+        uses `limit`/`offset`, unlike the `jobs` endpoint's page_size/page_offset)."""
         # Single-method dispatch today; when new methods are added,
         # branch on metadata.method to pick the concrete future type.
         return [
@@ -188,7 +185,7 @@ class ClusteringAPI:
             for md in api.clustering_list_get(
                 self.session,
                 method=method,
-                page_size=page_size,
-                page_offset=page_offset,
+                limit=limit,
+                offset=offset,
             )
         ]