arthur-client 1.1.0__tar.gz → 1.2.0__tar.gz

{arthur-client-1.1.0 → arthur-client-1.2.0}/PKG-INFO

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

{arthur-client-1.1.0 → arthur-client-1.2.0}/arthur/client/rest/admin/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  admin.yaml
-#   timestamp: 2024-01-16T21:33:56+00:00
+#   timestamp: 2024-01-19T17:36:04+00:00
 
 from __future__ import annotations
 

{arthur-client-1.1.0 → arthur-client-1.2.0}/arthur/client/rest/alerts/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  alerts.yaml
-#   timestamp: 2024-01-16T21:33:56+00:00
+#   timestamp: 2024-01-19T17:36:04+00:00
 
 from __future__ import annotations
 

{arthur-client-1.1.0 → arthur-client-1.2.0}/arthur/client/rest/bench/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  bench.yaml
-#   timestamp: 2024-01-16T21:33:55+00:00
+#   timestamp: 2024-01-19T17:36:03+00:00
 
 from __future__ import annotations
 

{arthur-client-1.1.0 → arthur-client-1.2.0}/arthur/client/rest/common/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  common.yaml
-#   timestamp: 2024-01-16T21:33:55+00:00
+#   timestamp: 2024-01-19T17:36:03+00:00
 
 from __future__ import annotations
 

{arthur-client-1.1.0 → arthur-client-1.2.0}/arthur/client/rest/enrichments/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  enrichments.yaml
-#   timestamp: 2024-01-16T21:33:55+00:00
+#   timestamp: 2024-01-19T17:36:03+00:00
 
 from __future__ import annotations
 

{arthur-client-1.1.0 → arthur-client-1.2.0}/arthur/client/rest/inferences/client.py

@@ -59,12 +59,14 @@ class ArthurInferencesClient:
         )
         return InferencesResponse(**parsed_resp)
 
-    def update_inference_ground_truths(
+    def update_inferences(
         self, model_id: str, json_body: List["PartnerIdInferenceGroundTruth"]
     ) -> InferencesResponse:
         """
         Updates inferences with ground truth data
 
+        This endpoint behaves similarly to the PATCH /v3/models/{model_id}/ground_truth endpoint, but it requires inference writing permissions.
+
         :param model_id:
         :param json_body: List of inference ground truth data
         """
@@ -76,6 +78,25 @@ class ArthurInferencesClient:
         )
         return InferencesResponse(**parsed_resp)
 
+    def update_inference_ground_truths(
+        self, model_id: str, json_body: List["PartnerIdInferenceGroundTruth"]
+    ) -> InferencesResponse:
+        """
+        Updates inferences with ground truth data
+
+        This endpoint behaves similarly to the PATCH /v3/models/{model_id}/inferences endpoint, but it requires ground truth writing permissions instead of inference writing permissions.
+
+        :param model_id:
+        :param json_body: List of inference ground truth data
+        """
+
+        parsed_resp: Dict[str, Any] = self.http_client.patch(  # type: ignore
+            f"/v3/models/{model_id}/ground_truth",
+            json=[x.dict(by_alias=True, exclude_none=True) for x in json_body],
+            validation_response_code=207,
+        )
+        return InferencesResponse(**parsed_resp)
+
     def send_inference_file(
         self, model_id: str, multipart_data: FileInferenceData
     ) -> ReferenceDataResponse:

{arthur-client-1.1.0 → arthur-client-1.2.0}/arthur/client/rest/inferences/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  inferences.yaml
-#   timestamp: 2024-01-16T21:33:55+00:00
+#   timestamp: 2024-01-19T17:36:03+00:00
 
 from __future__ import annotations
 

{arthur-client-1.1.0 → arthur-client-1.2.0}/arthur/client/rest/insights/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  insights.yaml
-#   timestamp: 2024-01-16T21:33:55+00:00
+#   timestamp: 2024-01-19T17:36:03+00:00
 
 from __future__ import annotations
 

{arthur-client-1.1.0 → arthur-client-1.2.0}/arthur/client/rest/metrics/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  metrics.yaml
-#   timestamp: 2024-01-16T21:33:55+00:00
+#   timestamp: 2024-01-19T17:36:03+00:00
 
 from __future__ import annotations
 

{arthur-client-1.1.0 → arthur-client-1.2.0}/arthur/client/rest/model_groups/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  model_groups.yaml
-#   timestamp: 2024-01-16T21:33:55+00:00
+#   timestamp: 2024-01-19T17:36:03+00:00
 
 from __future__ import annotations
 

{arthur-client-1.1.0 → arthur-client-1.2.0}/arthur/client/rest/models/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  models.yaml
-#   timestamp: 2024-01-16T21:33:55+00:00
+#   timestamp: 2024-01-19T17:36:03+00:00
 
 from __future__ import annotations
 

{arthur-client-1.1.0 → arthur-client-1.2.0}/arthur/client/rest/query/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  query.yaml
-#   timestamp: 2024-01-16T21:33:55+00:00
+#   timestamp: 2024-01-19T17:36:03+00:00
 
 from __future__ import annotations
 

{arthur-client-1.1.0 → arthur-client-1.2.0}/arthur/client/rest/users/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  users.yaml
-#   timestamp: 2024-01-16T21:33:55+00:00
+#   timestamp: 2024-01-19T17:36:03+00:00
 
 from __future__ import annotations
 

{arthur-client-1.1.0 → arthur-client-1.2.0}/arthur/client/version.py

@@ -1,2 +1,2 @@
-__version__ = "1.1.0"
+__version__ = "1.2.0"
 api_spec_version = "3.0.0"

{arthur-client-1.1.0 → arthur-client-1.2.0}/arthur_client.egg-info/PKG-INFO

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

{arthur-client-1.1.0 → arthur-client-1.2.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-1.1.0 → arthur-client-1.2.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"