arthur-client 0.11.0__tar.gz → 0.12.0__tar.gz

{arthur-client-0.11.0 → arthur-client-0.12.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arthur-client
-Version: 0.11.0
+Version: 0.12.0
 Summary: Arthur API Python Client
 Author-email: Arthur <info@arthur.ai>
 License: MIT

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur/client/rest/admin/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  admin.yaml
-#   timestamp: 2023-12-15T18:09:28+00:00
+#   timestamp: 2023-12-21T22:07:15+00:00
 
 from __future__ import annotations
 

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur/client/rest/alerts/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  alerts.yaml
-#   timestamp: 2023-12-15T18:09:28+00:00
+#   timestamp: 2023-12-21T22:07:15+00:00
 
 from __future__ import annotations
 

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur/client/rest/bench/client.py

@@ -42,7 +42,8 @@ class ArthurBenchClient:
         self,
         name: Optional[str] = None,
         sort: Optional[str] = None,
-        scoring_method: Optional[str] = None,
+        page: Optional[int] = None,
+        page_size: Optional[int] = None,
     ) -> PaginatedGetTestSuitesResponse:
         """
                 Gets test suites
@@ -52,7 +53,8 @@ class ArthurBenchClient:
 
                 :param name:
                 :param sort:
-                :param scoring_method:
+                :param page:
+                :param page_size:
         """
 
         params: Dict[str, Any] = {}
@@ -60,8 +62,10 @@ class ArthurBenchClient:
             params["name"] = name
         if sort is not None:
             params["sort"] = sort
-        if scoring_method is not None:
-            params["scoring_method"] = scoring_method
+        if page is not None:
+            params["page"] = page
+        if page_size is not None:
+            params["page_size"] = page_size
 
         parsed_resp: Dict[str, Any] = self.http_client.get(  # type: ignore
             f"/v3/bench/test_suites", params=params, validation_response_code=200
@@ -82,15 +86,30 @@ class ArthurBenchClient:
         )
         return TestSuiteResponse(**parsed_resp)
 
-    def get_test_suite(self, test_suite_id: str) -> PaginatedGetTestSuiteResponse:
+    def get_test_suite(
+        self,
+        test_suite_id: str,
+        page: Optional[int] = None,
+        page_size: Optional[int] = None,
+    ) -> PaginatedGetTestSuiteResponse:
         """
         Get reference data for an existing test suite
 
         :param test_suite_id:
+        :param page:
+        :param page_size:
         """
 
+        params: Dict[str, Any] = {}
+        if page is not None:
+            params["page"] = page
+        if page_size is not None:
+            params["page_size"] = page_size
+
         parsed_resp: Dict[str, Any] = self.http_client.get(  # type: ignore
-            f"/v3/bench/test_suites/{test_suite_id}", validation_response_code=200
+            f"/v3/bench/test_suites/{test_suite_id}",
+            params=params,
+            validation_response_code=200,
         )
         return PaginatedGetTestSuiteResponse(**parsed_resp)
 
@@ -118,7 +137,7 @@ class ArthurBenchClient:
         page_size: Optional[int] = None,
     ) -> TestSuiteSummaryResponse:
         """
-        Get pagianted summary statistics of a test suite
+        Get paginated summary statistics of a test suite
 
         Defaults to page size of 5.
 
@@ -144,18 +163,28 @@ class ArthurBenchClient:
         return TestSuiteSummaryResponse(**parsed_resp)
 
     def get_runs_for_test_suite(
-        self, test_suite_id: str, sort: Optional[str] = None
+        self,
+        test_suite_id: str,
+        sort: Optional[str] = None,
+        page: Optional[int] = None,
+        page_size: Optional[int] = None,
     ) -> PaginatedGetRunsForTestSuiteResponse:
         """
         Get runs for a particular test suite (identified by test_suite_id)
 
         :param test_suite_id:
         :param sort:
+        :param page:
+        :param page_size:
         """
 
         params: Dict[str, Any] = {}
         if sort is not None:
             params["sort"] = sort
+        if page is not None:
+            params["page"] = page
+        if page_size is not None:
+            params["page_size"] = page_size
 
         parsed_resp: Dict[str, Any] = self.http_client.get(  # type: ignore
             f"/v3/bench/test_suites/{test_suite_id}/runs",

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur/client/rest/bench/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  bench.yaml
-#   timestamp: 2023-12-15T18:09:28+00:00
+#   timestamp: 2023-12-21T22:07:15+00:00
 
 from __future__ import annotations
 

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur/client/rest/common/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  common.yaml
-#   timestamp: 2023-12-15T18:09:27+00:00
+#   timestamp: 2023-12-21T22:07:15+00:00
 
 from __future__ import annotations
 

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur/client/rest/enrichments/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  enrichments.yaml
-#   timestamp: 2023-12-15T18:09:27+00:00
+#   timestamp: 2023-12-21T22:07:14+00:00
 
 from __future__ import annotations
 

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur/client/rest/inferences/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  inferences.yaml
-#   timestamp: 2023-12-15T18:09:27+00:00
+#   timestamp: 2023-12-21T22:07:14+00:00
 
 from __future__ import annotations
 

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur/client/rest/insights/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  insights.yaml
-#   timestamp: 2023-12-15T18:09:27+00:00
+#   timestamp: 2023-12-21T22:07:14+00:00
 
 from __future__ import annotations
 

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur/client/rest/metrics/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  metrics.yaml
-#   timestamp: 2023-12-15T18:09:27+00:00
+#   timestamp: 2023-12-21T22:07:14+00:00
 
 from __future__ import annotations
 

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur/client/rest/model_groups/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  model_groups.yaml
-#   timestamp: 2023-12-15T18:09:27+00:00
+#   timestamp: 2023-12-21T22:07:14+00:00
 
 from __future__ import annotations
 

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur/client/rest/models/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  models.yaml
-#   timestamp: 2023-12-15T18:09:27+00:00
+#   timestamp: 2023-12-21T22:07:14+00:00
 
 from __future__ import annotations
 
@@ -503,6 +503,10 @@ class ModelResponse(ModelObject):
     """
     The total number of critical alerts for the model that are active (status is new or acknowledged).
     """
+    version_sequence_num: Optional[int] = None
+    """
+    An integer representing the sequence number of the model within its model group
+    """
 
 
 class ModelUpdateRequest(BaseModel):

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur/client/rest/query/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  query.yaml
-#   timestamp: 2023-12-15T18:09:27+00:00
+#   timestamp: 2023-12-21T22:07:14+00:00
 
 from __future__ import annotations
 

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur/client/version.py

@@ -1,2 +1,2 @@
-__version__ = "0.11.0"
+__version__ = "0.12.0"
 api_spec_version = "3.0.0"

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arthur-client
-Version: 0.11.0
+Version: 0.12.0
 Summary: Arthur API Python Client
 Author-email: Arthur <info@arthur.ai>
 License: MIT

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur/client/rest/model_groups/client.py

@@ -6,13 +6,13 @@ from requests import Response
 # import http client
 from arthur.client.http.requests import HTTPClient
 
-from arthur.client.rest.models.models import ModelExpand, ModelObject
 from arthur.client.rest.model_groups.models import (
     ModelGroupResponse,
     ModelGroupUpdateRequest,
     PaginatedModelGroupResponse,
     PaginatedModelGroupVersionsResponse,
 )
+from arthur.client.rest.models.models import ModelExpand, ModelObject
 
 
 PATH_PREFIX = "/api"

{arthur-client-0.11.0 → arthur-client-0.12.0}/arthur/client/rest/query/client.py

@@ -6,6 +6,7 @@ from requests import Response
 # import http client
 from arthur.client.http.requests import HTTPClient
 
+from arthur.client.rest.common.models import DataDriftRequest, QueryRequest
 from arthur.client.rest.query.models import (
     DataDriftResponse,
     DataDriftTableRequest,
@@ -13,7 +14,6 @@ from arthur.client.rest.query.models import (
     DistributionsRequest,
     QueryResult,
 )
-from arthur.client.rest.common.models import DataDriftRequest, QueryRequest
 
 
 PATH_PREFIX = "/api"