arthur-client 0.11.0__py3-none-any.whl → 0.12.0__py3-none-any.whl

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/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/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/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/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/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/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/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/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/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/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/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/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"

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/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.dist-info → arthur_client-0.12.0.dist-info}/METADATA

@@ -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.dist-info → arthur_client-0.12.0.dist-info}/RECORD

@@ -1,7 +1,7 @@
 arthur/client/__init__.py,sha256=hLwS66smzK1qUEyhyvXqZ_2L47IKkM7fjVNj7L2d24k,57
 arthur/client/helpers.py,sha256=toTAqb-r8SuvXXyy4ohef_2DOEyp8V7SidXL9W3gbxA,1457
 arthur/client/types.py,sha256=GufCfBBWoTe9MiH2NVSELgh-kJOvuj1-SiPcW0LKqGI,234
-arthur/client/version.py,sha256=sbc4FJ6FK_NHgJVV6uv4opsbtBv8on0flCsQhD1JRGE,50
+arthur/client/version.py,sha256=mFKpS7fIeL6M_TEHV-AoxGSOsgTHGdpzYuieCHTFnIw,50
 arthur/client/auth/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 arthur/client/auth/helpers.py,sha256=IWJ-s_yuVd0MMkQuA8S5njQdcAhMYoi8QMS7ksxAcnA,4881
 arthur/client/auth/refresh.py,sha256=Zn56iptWjnLGT0DIRsBSoRIBUPOB-7dLqVG-FZYW9yo,2681
@@ -13,43 +13,43 @@ arthur/client/rest/__init__.py,sha256=NiYncfs-JAR8LxTUGx4c2yBFWudotQv7FjDQNpjvYY
 arthur/client/rest/client.py,sha256=FYCi6RI0o2HbjZG2UufidJxBd8H-cLT68O1Glmmf3QY,6857
 arthur/client/rest/admin/__init__.py,sha256=FdmrkBO-lbpLjMuaRZBDGhEnapOBYLP6U7j0-PqKpKw,42
 arthur/client/rest/admin/client.py,sha256=70FjNlxt1-QQmE_7m5Qux3WfTtOehO4VgJJx01L80IY,14999
-arthur/client/rest/admin/models.py,sha256=kixBCRVnHpi-_iXFuT-ENBdn8z05CT8S6GYSJViHV10,22897
+arthur/client/rest/admin/models.py,sha256=kqVzSqXIr1Zc7xpWM9qmLfbKKA3vdBy7iI-Jy5AzMAg,22897
 arthur/client/rest/alerts/__init__.py,sha256=FdmrkBO-lbpLjMuaRZBDGhEnapOBYLP6U7j0-PqKpKw,42
 arthur/client/rest/alerts/client.py,sha256=-RAhKM05GUhw307b-f0ept4Fy6n3SwF3onXbAYCXLA8,20399
-arthur/client/rest/alerts/models.py,sha256=apY7jfLpJngoPE7HfLRYjSxnbz9hH5oPvvJydmRokIg,19582
+arthur/client/rest/alerts/models.py,sha256=FlCuPvkGfChY0BcCxRwQbcin6IPalYfoYUU778AWam8,19582
 arthur/client/rest/bench/__init__.py,sha256=FdmrkBO-lbpLjMuaRZBDGhEnapOBYLP6U7j0-PqKpKw,42
-arthur/client/rest/bench/client.py,sha256=den3fwJ_xrnUfq1F5vsRsGzdFOZIkO8NNbB7fklOXWI,7808
-arthur/client/rest/bench/models.py,sha256=HLYI5wZD5n_PGVTovgtPfrDOPBkLbvdjnMmjsL_Zyjk,5165
+arthur/client/rest/bench/client.py,sha256=eRVemQiqYvXqntjKzyICdvrPgbzXD780LXDfHHQBYRs,8558
+arthur/client/rest/bench/models.py,sha256=M5zFGtRTzNZdAr0WSy7xqT0GcYQ6IIGql3jpr-MvBf0,5165
 arthur/client/rest/common/__init__.py,sha256=FdmrkBO-lbpLjMuaRZBDGhEnapOBYLP6U7j0-PqKpKw,42
 arthur/client/rest/common/client.py,sha256=wd24QtCkGKvzbWVhKxwUYxZsoaN6YU-NheQIv6k6YHo,701
-arthur/client/rest/common/models.py,sha256=j4824-xOnOoytN1l0Wp1rcdSt670g368VwW8zd7VDBc,10006
+arthur/client/rest/common/models.py,sha256=QJzLxNgxJamk0tyTfpUIPE0Xeg_LVDhTI8STrn0gKjY,10006
 arthur/client/rest/enrichments/__init__.py,sha256=FdmrkBO-lbpLjMuaRZBDGhEnapOBYLP6U7j0-PqKpKw,42
 arthur/client/rest/enrichments/client.py,sha256=U8LOVm_MXZMqmtoqx6ASEK2Nl-19CZbEyEQlPqpl4xI,12252
-arthur/client/rest/enrichments/models.py,sha256=QE5Js3GlnHqXNiDXWFYVRkRHZz1zuIvS2XfpplX-lf8,14382
+arthur/client/rest/enrichments/models.py,sha256=A-cF7aqC-78McNQf7Q8AxVMhLcpGhtI2dz3cbGcTGE8,14382
 arthur/client/rest/inferences/__init__.py,sha256=FdmrkBO-lbpLjMuaRZBDGhEnapOBYLP6U7j0-PqKpKw,42
 arthur/client/rest/inferences/client.py,sha256=YY56LBzjHrGCyrUf9ETy7I7D6V9gNtOXBfodjFsPJB8,9071
-arthur/client/rest/inferences/models.py,sha256=TU6nC0XbrUlZ1bp8bszTTZRG1DF3w8mT2SSGMOcj92U,12571
+arthur/client/rest/inferences/models.py,sha256=BgcO0DvYbp9YSTRL-aaOijZnoMA2OdgcZxeiALo1Ovw,12571
 arthur/client/rest/insights/__init__.py,sha256=FdmrkBO-lbpLjMuaRZBDGhEnapOBYLP6U7j0-PqKpKw,42
 arthur/client/rest/insights/client.py,sha256=8ok5oDzdBeytwMAjuSuyTI8xh_C72w1czo3mpxO7GVw,6364
-arthur/client/rest/insights/models.py,sha256=vezzZeUU8MaoCG8yGB8yac71ko_sS8Hu9pRPJSrZRFA,4082
+arthur/client/rest/insights/models.py,sha256=Yq2glbkl-SdCROXT8SFTKxhCkTz1Gj0uoDDudbn1gJM,4082
 arthur/client/rest/metrics/__init__.py,sha256=FdmrkBO-lbpLjMuaRZBDGhEnapOBYLP6U7j0-PqKpKw,42
 arthur/client/rest/metrics/client.py,sha256=jjynBs2qi1b0x3wGwruKpDn9RJ_FuTZ3Y54kjh-eme0,9819
-arthur/client/rest/metrics/models.py,sha256=zgTdshpCrvIdIKnpPB0J3ApyjvgHqGy1jyY8fKa_7gk,8267
+arthur/client/rest/metrics/models.py,sha256=HQsNKG8ysrDm3ZwY5bwbBVSWTNiM_XlWOsmtmx3_f_k,8267
 arthur/client/rest/model_groups/__init__.py,sha256=FdmrkBO-lbpLjMuaRZBDGhEnapOBYLP6U7j0-PqKpKw,42
-arthur/client/rest/model_groups/client.py,sha256=2F50s9yXP1bX1ohDo216TYqMYXBXHxjtwJyWXESBxU8,5020
-arthur/client/rest/model_groups/models.py,sha256=AVCRK5s1PYOU3bc93JHywJm510YqV9WE91IFU1A39Sc,2151
+arthur/client/rest/model_groups/client.py,sha256=A4_n_G-dejm4Ln5d6qAS83I6xAqz9qbUn0SXteoTxhI,5020
+arthur/client/rest/model_groups/models.py,sha256=cAby6NSkC6wBwhNjyzYglBB4g-xn7vSo5uGyq-6k76U,2151
 arthur/client/rest/models/__init__.py,sha256=FdmrkBO-lbpLjMuaRZBDGhEnapOBYLP6U7j0-PqKpKw,42
 arthur/client/rest/models/client.py,sha256=KUGdTDYbAMjzHhvMPoMKc05m4U15CCslwi5uWRrWJHI,16190
-arthur/client/rest/models/models.py,sha256=TCGU_gO0Zce1-K7BIt_nl88Mu7rS5LwMPhM5s2uu88A,16469
+arthur/client/rest/models/models.py,sha256=Qf6LCgJQ3et3KBIwP1Yj1GxLlXhAIaEsYvbknjkwQQk,16616
 arthur/client/rest/query/__init__.py,sha256=FdmrkBO-lbpLjMuaRZBDGhEnapOBYLP6U7j0-PqKpKw,42
-arthur/client/rest/query/client.py,sha256=EIpEqcn7m-a3Ku3KO95NINdx83ggfOmGdin0Z0oHH3E,4125
-arthur/client/rest/query/models.py,sha256=QsD1glnjDQJV_p_88iJ2WGm5BGKxPjkcaIY4C9Dnmm0,4223
+arthur/client/rest/query/client.py,sha256=A8ZpF34G9koOf3_H_mqyNiCRk-gC8p5Vp4x6AhEnIww,4125
+arthur/client/rest/query/models.py,sha256=PwKKeuQ5W_HLkzzES4Uqv-Ls9oi-CfeCQoEKr0m5Hq8,4223
 arthur/common/__init__.py,sha256=ZPrXCVGLk4xZJQhuxYHKQYWUOpa3uHsqhf0T8qzgpjQ,68
 arthur/common/constants.py,sha256=4_o_WDbtoOJ7kANCV28HdFbDJDO90k2TGULHxjqeSao,159
 arthur/common/exceptions.py,sha256=m_lGhrb-EoMZCq6XjYPJLQEfAhYkE7h0Wuz0bNMtV4c,5230
 arthur/common/log.py,sha256=oDB_teFV-pdDfcM3VGSSWYSOu6Iag_3RyzRDWBcIcOM,2851
-arthur_client-0.11.0.dist-info/LICENSE,sha256=a0qChOPgo0vduazUlCCpPkJ4A7nOtvBTbbvNh6qDmhw,1063
-arthur_client-0.11.0.dist-info/METADATA,sha256=l_SWNo2QzqKTV-ohJgHbSAj--3FkroDtRTvh9lSnJEk,1738
-arthur_client-0.11.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-arthur_client-0.11.0.dist-info/top_level.txt,sha256=u8s5CFDcE7LoZePEhTOipwnPqeiXh4_hYXJp7vg5BLQ,7
-arthur_client-0.11.0.dist-info/RECORD,,
+arthur_client-0.12.0.dist-info/LICENSE,sha256=a0qChOPgo0vduazUlCCpPkJ4A7nOtvBTbbvNh6qDmhw,1063
+arthur_client-0.12.0.dist-info/METADATA,sha256=ZCstqmm7ys1UdANLD2Mn1k0atZ6ZoRUOvhS42SG1IJY,1738
+arthur_client-0.12.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+arthur_client-0.12.0.dist-info/top_level.txt,sha256=u8s5CFDcE7LoZePEhTOipwnPqeiXh4_hYXJp7vg5BLQ,7
+arthur_client-0.12.0.dist-info/RECORD,,