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

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

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

{arthur-client-0.10.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-11T22:24:33+00:00
+#   timestamp: 2023-12-21T22:07:15+00:00
 
 from __future__ import annotations
 

{arthur-client-0.10.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-11T22:24:33+00:00
+#   timestamp: 2023-12-21T22:07:15+00:00
 
 from __future__ import annotations
 

{arthur-client-0.10.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)
 
@@ -113,24 +132,24 @@ class ArthurBenchClient:
     def get_summary_statistics(
         self,
         test_suite_id: str,
-        run_id: Optional[str] = None,
+        run_ids: Optional[List[str]] = None,
         page: Optional[int] = None,
         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.
 
         :param test_suite_id:
-        :param run_id:
+        :param run_ids:
         :param page:
         :param page_size:
         """
 
         params: Dict[str, Any] = {}
-        if run_id is not None:
-            params["run_id"] = run_id
+        if run_ids is not None:
+            params["run_ids"] = run_ids
         if page is not None:
             params["page"] = page
         if page_size is not None:
@@ -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.10.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-11T22:24:33+00:00
+#   timestamp: 2023-12-21T22:07:15+00:00
 
 from __future__ import annotations
 

{arthur-client-0.10.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-11T22:24:33+00:00
+#   timestamp: 2023-12-21T22:07:15+00:00
 
 from __future__ import annotations
 
@@ -301,7 +301,7 @@ class MetricQuery(BaseModel):
     This should be in the format described in our [Query Guide](/api-documentation/query_guide.html). Your query must have exactly one select and may not use the `group_by` or `subquery` fields. Don't include any batch or timestamp filters as these will be added dynamically. For details [see the full metrics docs](https://docs.arthur.ai/user-guide/walkthroughs/metrics_alerts.html#creating-custom-metrics).
     """
 
-    select: QuerySelect
+    select: Optional[QuerySelect] = None
     from_: Optional[From1] = Field(None, alias='from')
     """
     Data to query. Inferences are the raw inferences sent to the platform. Enriched includes the raw inferences, plus enriched properties like explanations and anomaly scores. Reference is the reference data set uploaded to the platform. The default is 'inference'.

{arthur-client-0.10.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-11T22:24:33+00:00
+#   timestamp: 2023-12-21T22:07:14+00:00
 
 from __future__ import annotations
 

{arthur-client-0.10.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-11T22:24:33+00:00
+#   timestamp: 2023-12-21T22:07:14+00:00
 
 from __future__ import annotations
 

{arthur-client-0.10.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-11T22:24:33+00:00
+#   timestamp: 2023-12-21T22:07:14+00:00
 
 from __future__ import annotations
 

{arthur-client-0.10.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-11T22:24:33+00:00
+#   timestamp: 2023-12-21T22:07:14+00:00
 
 from __future__ import annotations
 

{arthur-client-0.10.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-11T22:24:32+00:00
+#   timestamp: 2023-12-21T22:07:14+00:00
 
 from __future__ import annotations
 

{arthur-client-0.10.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-11T22:24:32+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.10.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-11T22:24:32+00:00
+#   timestamp: 2023-12-21T22:07:14+00:00
 
 from __future__ import annotations
 

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

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

{arthur-client-0.10.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.10.0
+Version: 0.12.0
 Summary: Arthur API Python Client
 Author-email: Arthur <info@arthur.ai>
 License: MIT

{arthur-client-0.10.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.10.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"