arthur-client 0.5.0__tar.gz → 0.6.0__tar.gz

{arthur-client-0.5.0 → arthur-client-0.6.0}/PKG-INFO

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

{arthur-client-0.5.0 → arthur-client-0.6.0}/arthur/client/rest/admin/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  admin.yaml
-#   timestamp: 2023-10-03T17:05:09+00:00
+#   timestamp: 2023-10-04T15:04:36+00:00
 
 from __future__ import annotations
 

{arthur-client-0.5.0 → arthur-client-0.6.0}/arthur/client/rest/alerts/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  alerts.yaml
-#   timestamp: 2023-10-03T17:05:08+00:00
+#   timestamp: 2023-10-04T15:04:36+00:00
 
 from __future__ import annotations
 

{arthur-client-0.5.0 → arthur-client-0.6.0}/arthur/client/rest/bench/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  bench.yaml
-#   timestamp: 2023-10-03T17:05:08+00:00
+#   timestamp: 2023-10-04T15:04:36+00:00
 
 from __future__ import annotations
 

{arthur-client-0.5.0 → arthur-client-0.6.0}/arthur/client/rest/common/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  common.yaml
-#   timestamp: 2023-10-03T17:05:08+00:00
+#   timestamp: 2023-10-04T15:04:36+00:00
 
 from __future__ import annotations
 

{arthur-client-0.5.0 → arthur-client-0.6.0}/arthur/client/rest/enrichments/client.py

@@ -12,7 +12,7 @@ from arthur.client.rest.enrichments.models import (
     BiasConstraintEnum,
     BiasMitigationEnrichmentConfiguration,
     BiasMitigationEnrichmentResponse,
-    EnrichmentsConfiguration,
+    EnrichmentsRequest,
     EnrichmentsResponse,
     ExplainabilityEnrichmentMultipartRequestBody,
     ExplainabilityEnrichmentRequest,
@@ -102,14 +102,14 @@ class ArthurEnrichmentsClient:
     def update_enrichment_config(
         self,
         model_id: str,
-        json_body: EnrichmentsConfiguration,
+        json_body: EnrichmentsRequest,
         multipart_data: ExplainabilityEnrichmentMultipartRequestBody,
     ) -> Response:
         """
         Updates the enrichment configuration for a model
 
         :param model_id:
-        :param json_body: A JSON-formatted enrichments configuration. See the EnrichmentsConfiguration object for schema
+        :param json_body: Configures multiple enrichments. A multipart/form-data body with at least a `configuration` JSON body. If explainability is being enabled for the first time, artifacts must be supplied.
         :param multipart_data: When setting up explainability, a config must always be provided. The explainability enrichment artifact files may be provided all together, but a config must be provided as well, regardless of whether the config has already been set.
         """
 

{arthur-client-0.5.0 → arthur-client-0.6.0}/arthur/client/rest/enrichments/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  enrichments.yaml
-#   timestamp: 2023-10-03T17:05:08+00:00
+#   timestamp: 2023-10-04T15:04:36+00:00
 
 from __future__ import annotations
 
@@ -199,6 +199,17 @@ class ExplainabilityEnrichmentRequest(BaseModel):
     """
 
 
+class EnrichmentsRequest(BaseModel):
+    """
+    Configures multiple enrichments. A multipart/form-data body with at least a `configuration` JSON body. If explainability is being enabled for the first time, artifacts must be supplied.
+    """
+
+    config: Optional[EnrichmentsConfiguration] = None
+    """
+    Enrichments configuration
+    """
+
+
 class ExplanationValuesWhatIf(BaseModel):
     attribute_name: str = Field(..., example='feature_a')
     explanation_value: float = Field(..., example=0.12)
@@ -333,14 +344,6 @@ class WhatIfRequest(BaseModel):
     model_pipeline_input: List[WhatIfAttributeRequest]
 
 
-class EnrichmentsRequest(EnrichmentsConfiguration):
-    """
-    Configures multiple enrichments. A multipart/form-data body with at least a `configuration` JSON body. If explainability is being enabled for the first time, artifacts must be supplied.
-    """
-
-    pass
-
-
 class EnrichmentStatusUpdate(EnrichmentStatus):
     """
     Updates status for an enrichment. A body with at least a `status` key.

{arthur-client-0.5.0 → arthur-client-0.6.0}/arthur/client/rest/inferences/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  inferences.yaml
-#   timestamp: 2023-10-03T17:05:08+00:00
+#   timestamp: 2023-10-04T15:04:36+00:00
 
 from __future__ import annotations
 

{arthur-client-0.5.0 → arthur-client-0.6.0}/arthur/client/rest/insights/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  insights.yaml
-#   timestamp: 2023-10-03T17:05:08+00:00
+#   timestamp: 2023-10-04T15:04:35+00:00
 
 from __future__ import annotations
 

{arthur-client-0.5.0 → arthur-client-0.6.0}/arthur/client/rest/metrics/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  metrics.yaml
-#   timestamp: 2023-10-03T17:05:08+00:00
+#   timestamp: 2023-10-04T15:04:35+00:00
 
 from __future__ import annotations
 

{arthur-client-0.5.0 → arthur-client-0.6.0}/arthur/client/rest/model_groups/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  model_groups.yaml
-#   timestamp: 2023-10-03T17:05:08+00:00
+#   timestamp: 2023-10-04T15:04:35+00:00
 
 from __future__ import annotations
 

{arthur-client-0.5.0 → arthur-client-0.6.0}/arthur/client/rest/models/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  models.yaml
-#   timestamp: 2023-10-03T17:05:08+00:00
+#   timestamp: 2023-10-04T15:04:35+00:00
 
 from __future__ import annotations
 

{arthur-client-0.5.0 → arthur-client-0.6.0}/arthur/client/rest/query/models.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  query.yaml
-#   timestamp: 2023-10-03T17:05:08+00:00
+#   timestamp: 2023-10-04T15:04:35+00:00
 
 from __future__ import annotations
 

{arthur-client-0.5.0 → arthur-client-0.6.0}/arthur/client/version.py

@@ -1,2 +1,2 @@
-__version__ = "0.5.0"
+__version__ = "0.6.0"
 api_spec_version = "3.0.0"

{arthur-client-0.5.0 → arthur-client-0.6.0}/arthur_client.egg-info/PKG-INFO

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

{arthur-client-0.5.0 → arthur-client-0.6.0}/arthur/client/rest/query/client.py

@@ -6,7 +6,6 @@ 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,
@@ -14,6 +13,7 @@ from arthur.client.rest.query.models import (
     DistributionsRequest,
     QueryResult,
 )
+from arthur.client.rest.common.models import DataDriftRequest, QueryRequest
 
 
 PATH_PREFIX = "/api"